VideoSummarizer / backend /setup_model.py
malikKaleem's picture
Whisper Model process on video in backend
bc97bb3
Raw
History Blame Contribute Delete
563 Bytes
import os
import whisper
from config import settings
def setup_whisper():
"""Download the Whisper model to the local models directory."""
print(f"Downloading Whisper model '{settings.WHISPER_MODEL_TYPE}' to {settings.MODELS_DIR}...")
if not os.path.exists(settings.MODELS_DIR):
os.makedirs(settings.MODELS_DIR)
# This triggers the download
whisper.load_model(settings.WHISPER_MODEL_TYPE, download_root=settings.MODELS_DIR)
print("Model downloaded successfully!")
if __name__ == "__main__":
setup_whisper()