Update app.py

app.py

@@ -12,31 +12,31 @@ from huggingface_hub import hf_hub_download
 # 1. Define Hugging Face dataset + weights
 # =========================================
 
-HF_DATASET_REPO = "roll-ai/FloVD-weights"  # your dataset repo on HF
-
-WEIGHT_FILES = {
-    "ckpt/FVSM/FloVD_FVSM_Controlnet.pt": "FVSM/FloVD_FVSM_Controlnet.pt",
-    "ckpt/OMSM/selected_blocks.safetensors": "OMSM/selected_blocks.safetensors",
-    "ckpt/OMSM/pytorch_lora_weights.safetensors": "OMSM/pytorch_lora_weights.safetensors",
-    "ckpt/others/depth_anything_v2_metric_hypersim_vitb.pth": "others/depth_anything_v2_metric_hypersim_vitb.pth"
-}
-
-def download_weights():
-    print("🔄 Downloading model weights via huggingface_hub...")
-    for hf_path, local_rel_path in WEIGHT_FILES.items():
-        local_path = Path("ckpt") / local_rel_path
-        if not local_path.exists():
-            print(f"📥 Downloading {hf_path}")
-            hf_hub_download(
-                repo_id=HF_DATASET_REPO,
-                repo_type="dataset",
-                filename=hf_path,
-                local_dir="./"
-            )
-        else:
-            print(f"✅ Already exists: {local_path}")
-
-download_weights()
+# HF_DATASET_REPO = "roll-ai/FloVD-weights"  # your dataset repo on HF
+
+# WEIGHT_FILES = {
+#     "ckpt/FVSM/FloVD_FVSM_Controlnet.pt": "FVSM/FloVD_FVSM_Controlnet.pt",
+#     "ckpt/OMSM/selected_blocks.safetensors": "OMSM/selected_blocks.safetensors",
+#     "ckpt/OMSM/pytorch_lora_weights.safetensors": "OMSM/pytorch_lora_weights.safetensors",
+#     "ckpt/others/depth_anything_v2_metric_hypersim_vitb.pth": "others/depth_anything_v2_metric_hypersim_vitb.pth"
+# }
+
+# def download_weights():
+#     print("🔄 Downloading model weights via huggingface_hub...")
+#     for hf_path, local_rel_path in WEIGHT_FILES.items():
+#         local_path = Path("ckpt") / local_rel_path
+#         if not local_path.exists():
+#             print(f"📥 Downloading {hf_path}")
+#             hf_hub_download(
+#                 repo_id=HF_DATASET_REPO,
+#                 repo_type="dataset",
+#                 filename=hf_path,
+#                 local_dir="./"
+#             )
+#         else:
+#             print(f"✅ Already exists: {local_path}")
+
+# download_weights()
 
 # =========================================
 # 2. Import the FloVD generation pipeline