whisper-demo-es-medium / transcriber.py
juancopi81's picture
Add filepath
195e557
raw
history blame
441 Bytes
from transformers import pipeline
class Transcriber:
def __init__(self, pipe: pipeline) -> None:
self.pipe = pipe
def transcribe(self, file_path: str = "yt_audio.mp3") -> str:
print(self.pipe)
print("Audo file at:", file_path)
try:
transcription = self.pipe(file_path)["text"]
return transcription
except:
return "ERROR: No audio file found to transcribe"