Spaces:
Sleeping
Sleeping
| 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() | |