voice-to-code-backend / whisper_model.py
ShobhitKori
Updated whisper_model.py
4df0d17
raw
history blame
524 Bytes
import os
# Add FFmpeg directory to PATH
os.environ["PATH"] += os.pathsep + r"C:\ffmpeg\bin"
import whisper
import logging
model = whisper.load_model("small")
# def transcribe_audio(file_path: str) -> str:
# result = model.transcribe(file_path)
# return result["text"]
logging.basicConfig(level=logging.INFO)
def transcribe_audio(file_path):
logging.info("Starting transcription for %s", file_path)
result = model.transcribe(file_path)
logging.info("Transcription completed")
return result["text"]