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