AutomaticSpeechRecognition.py

import torch
from transformers import pipeline
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

class AutomaticSpeechRecognition():
    QuestionAudio = None
    ##
    QuestionText = None

    def funcAutomaticSpeechRecognition(self, input):
        '''
        Verify the inference
        '''
        speech_reco = pipeline(
            "automatic-speech-recognition", model="openai/whisper-base", device=device
        )
        res = speech_reco(input)
        return res["text"]

    def run(self):
        self.QuestionText = self.funcAutomaticSpeechRecognition(self.QuestionAudio)

if __name__ == '__main__':
    module = AutomaticSpeechRecognition()
    module.QuestionAudio = "path/to/audiofile"
    module.run()
    print(module.QuestionText)