download_model.py

from transformers import WhisperProcessor, WhisperForConditionalGeneration
import os

# Ensure the cache directory exists
cache_dir = "/app/.cache"
os.makedirs(cache_dir, exist_ok=True)

print("Downloading Whisper model and processor...")
try:
    processor = WhisperProcessor.from_pretrained("openai/whisper-large-v3", cache_dir=cache_dir)
    model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-large-v3", cache_dir=cache_dir)
    print("✓ Model and processor downloaded successfully!")
except Exception as e:
    print(f"Failed to download model: {e}")
    raise