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)