File size: 552 Bytes
a159b10
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
from transformers import pipeline

class SpeechRecognizer:
    def __init__(self, model_size="tiny"):
        print(f"Loading Whisper {model_size} model...")
        self.model = pipeline(
            "automatic-speech-recognition",
            model=f"openai/whisper-{model_size}",
            device=-1
        )
        print("Pipeline created successfully")

    def transcribe_audio(self, audio_path):
        result = self.model(
            audio_path,
            generate_kwargs={"language": "english"}
        )
        return result["text"]