Update app.py

app.py

@@ -15,11 +15,19 @@ os.environ["TORCH_HOME"] = cache_dir  # Set PyTorch cache directory
 # Ensure the cache directory exists and is writable
 os.makedirs(cache_dir, exist_ok=True)
 
-# Load the fine-tuned Macedonian-ASR Whisper model and processor
+# Load the base Whisper model and processor
 def load_model():
-    print("Loading fine-tuned Macedonian-ASR Whisper model and processor...")
-    processor = WhisperProcessor.from_pretrained("Macedonian-ASR/whisper-large-v3-macedonian-asr")
-    model = WhisperForConditionalGeneration.from_pretrained("Macedonian-ASR/whisper-large-v3-macedonian-asr")
+    print("Loading base Whisper model and processor...")
+    processor = WhisperProcessor.from_pretrained("openai/whisper-large-v3")
+    model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-large-v3")
+    
+    # Load the fine-tuned weights from the Macedonian-ASR repository
+    print("Loading fine-tuned weights from Macedonian-ASR...")
+    model.load_state_dict(torch.hub.load_state_dict_from_url(
+        "https://huggingface.co/Macedonian-ASR/whisper-large-v3-macedonian-asr/resolve/main/pytorch_model.bin",
+        map_location="cpu",
+        model_dir=cache_dir  # Save downloaded weights to the writable cache directory
+    ))
     print("✓ Model and processor loaded successfully!")
     return processor, model