Spaces:
Running
Running
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"]
|