Update app.py

app.py

@@ -50,15 +50,28 @@ head = MultimodalRegressor().to(DEVICE)
 
 # NEW: Dynamic load with cache
 def load_model_if_needed():
-    model_path = hf_hub_download(
-        repo_id="MeshMax/video_tower",
-        filename="finetuned_multimodal.pt",
-        local_dir="/tmp",
-        local_dir_use_symlinks=False,
-        cache_dir=None
-    )
-    print(f"Model loaded from: {model_path}")
-    return model_path
+    try:
+        model_path = hf_hub_download(
+            repo_id="MeshMax/video_tower",
+            filename="finetuned_multimodal.pt",
+            local_dir=None,  # CHANGED: Use default ~/.cache (persistent, no /tmp)
+            local_dir_use_symlinks=False,
+            cache_dir=None
+        )
+        print(f"Model loaded from: {model_path}")
+        return model_path
+    except Exception as e:
+        print(f"Download failed: {e}. Retrying with force_download...")
+        # Fallback: Force re-download if cache is corrupted
+        model_path = hf_hub_download(
+            repo_id="MeshMax/video_tower",
+            filename="finetuned_multimodal.pt",
+            local_dir=None,
+            local_dir_use_symlinks=False,
+            cache_dir=None,
+            force_download=True  # Overwrite if needed
+        )
+        return model_path
 
 model_path = load_model_if_needed()
 ckpt = torch.load(model_path, map_location=DEVICE, weights_only=False)