download checkpoints

app.py

@@ -1417,8 +1417,49 @@ def create_ui(
     return demo
 
 
+def download_required_files():
+    import subprocess
+
+    # 1. Download "checkpoints" directory from huggingface.co/gdhe17/Self-Forcing
+    if not os.path.exists("checkpoints"):
+        print("Downloading checkpoints from gdhe17/Self-Forcing...")
+        subprocess.run(
+            [
+                "huggingface-cli", "download",
+                "gdhe17/Self-Forcing",
+                "--include", "checkpoints/*",
+                "--local-dir", ".",
+            ],
+            check=True,
+        )
+
+    # 2. Download Wan-AI/Wan2.1-T2V-1.3B and place in wan_models/Wan2.1-T2V-1.3B
+    wan_model_dir = os.path.join("wan_models", "Wan2.1-T2V-1.3B")
+    if not os.path.exists(wan_model_dir):
+        print("Downloading Wan-AI/Wan2.1-T2V-1.3B...")
+        os.makedirs("wan_models", exist_ok=True)
+        subprocess.run(
+            [
+                "huggingface-cli", "download",
+                "Wan-AI/Wan2.1-T2V-1.3B",
+                "--local-dir", wan_model_dir,
+            ],
+            check=True,
+        )
+
+    # 3. Download SAM ViT-H checkpoint
+    sam_checkpoint_path = "sam_vit_h_4b8939.pth"
+    if not os.path.exists(sam_checkpoint_path):
+        print("Downloading SAM ViT-H checkpoint...")
+        import urllib.request
+        urllib.request.urlretrieve(
+            "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth",
+            sam_checkpoint_path,
+        )
+
 def main():
-    sam_model = sam_model_registry["vit_h"](checkpoint="../segment-anything/sam_vit_h_4b8939.pth")
+    download_required_files()
+    sam_model = sam_model_registry["vit_h"](checkpoint="./sam_vit_h_4b8939.pth")
     sam_model.to(device="cuda")
     sam_predictor = SamPredictor(sam_model)