Update download_models.py

download_models.py

@@ -1,22 +1,18 @@
+import os
 from huggingface_hub import snapshot_download
 
-# تحميل نموذج Wan2.1
-snapshot_download(
-    repo_id="Wan-AI/Wan2.1-I2V-14B-720P",
-    local_dir="./models/Wan2.1-I2V-14B-720P",
-    local_dir_use_symlinks=False
-)
+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
+        )
+    else:
+        print(f"✅ Model already exists at {local_dir}, skipping download.")
 
-# تحميل مشفّر الصوت wav2vec2
-snapshot_download(
-    repo_id="facebook/wav2vec2-base-960h",
-    local_dir="./models/wav2vec2-base-960h",
-    local_dir_use_symlinks=False
-)
-
-# تحميل موديل FantasyTalking
-snapshot_download(
-    repo_id="acvlab/FantasyTalking",
-    local_dir="./models/fantasytalking_model",
-    local_dir_use_symlinks=False
-)
+# تحميل النماذج فقط إذا ما كانت موجودة
+maybe_download_model("Wan-AI/Wan2.1-I2V-14B-720P", "./models/Wan2.1-I2V-14B-720P")
+maybe_download_model("facebook/wav2vec2-base-960h", "./models/wav2vec2-base-960h")
+maybe_download_model("acvlab/FantasyTalking", "./models/fantasytalking_model")