SpeechRecognition.py

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

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

    def funcSpeechRecognition(self, input):
        '''
        Verify the inference
        '''
        if self.QuestionText == None:
            playsound(input)
        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.funcSpeechRecognition(self.QuestionAudio)

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