Spaces:
Build error
Build error
from abc import ABC, abstractmethod | |
from transformers import pipeline | |
class Transcriber(ABC): | |
def transcribe(self, file_path: str) -> str: | |
pass | |
class SpanishTranscriber(Transcriber): | |
def __init__(self, pipe: pipeline) -> None: | |
self.pipe = pipe | |
def transcribe(self, file_path: str = "yt_audio.mp3") -> str: | |
print("Pipe:", self.pipe) | |
print("Audo file at:", file_path) | |
transcription = self.pipe(file_path)["text"] | |
return transcription | |
class WhisperTranscriber(Transcriber): | |
def __init__(self, model, without_timestamps: bool=True) -> None: | |
self.model = model | |
self.without_timestamps = without_timestamps | |
def transcribe(self, file_path: str = "yt_audio.mp3") -> str: | |
print("Model:", self.model) | |
print("Audo file at:", file_path) | |
transcription = self.model.transcribe(file_path, | |
without_timestamps=self.without_timestamps)["text"] | |
return transcription |