adityaardak/office

1import whisper2import torch3 4class SpeechToText:5    def __init__(self, model_size="base"):6        device = "cuda" if torch.cuda.is_available() else "cpu"7        print(f"Loading Whisper '{model_size}' model on {device}...")8        self.model = whisper.load_model(model_size, device=device)9        print("Whisper model loaded successfully.")10 11    def transcribe(self, audio_path):12        print("Starting transcription with Whisper...")13        result = self.model.transcribe(audio_path)14        transcript = result["text"]15        print("Transcription completed.")16        return transcript17