Update download_model.py

download_model.py

@@ -2,11 +2,14 @@ from transformers import WhisperProcessor, WhisperForConditionalGeneration
 import os
 
 # Ensure the cache directory exists
-cache_dir = "./.cache"
+cache_dir = "/app/.cache"
 os.makedirs(cache_dir, exist_ok=True)
 
-# Download the model and processor
 print("Downloading Whisper model and processor...")
-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!")
+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