Update download_models.py

download_models.py

@@ -2,13 +2,13 @@ import os
 from huggingface_hub import snapshot_download
 
 def maybe_download_model(repo_id, local_dir):
-    if not os.path.exists(local_dir) or not os.listdir(local_dir):
-        print(f"📥 Downloading model from {repo_id} ...")
-        snapshot_download(
-            repo_id=repo_id,
-            local_dir=local_dir,
-            local_dir_use_symlinks=False
-        )
+    if not os.path.exists(local_dir):
+        print(f"Downloading {repo_id} into {local_dir}")
+        snapshot_download(repo_id=repo_id, local_dir=local_dir, local_dir_use_symlinks=False)
     else:
-        print(f"✅ Model already exists at {local_dir}, skipping download.")
+        print(f"✅ {repo_id} already downloaded to {local_dir}")
+
+if __name__ == "__main__":
+    maybe_download_model("wedyanessam/fantasy-talking-model2", "./models/fantasytalking_model")
+    maybe_download_model("guoyww/wan-video", "./models/Wan2.1-I2V-14B-720P")