Update app.py

app.py

@@ -45,7 +45,6 @@ from huggingface_hub import snapshot_download
 # Constants
 MODEL_ZOO_DIR = "./model_zoo"
 PUSA_DIR = os.path.join(MODEL_ZOO_DIR, "PusaV1")
-WAN_DIR = os.path.join(MODEL_ZOO_DIR, "Wan2.1-T2V-14B")
 WAN_SUBFOLDER = "Wan2.1-T2V-14B"
 WAN_MODEL_PATH = os.path.join(PUSA_DIR, WAN_SUBFOLDER)
 LORA_PATH = os.path.join(PUSA_DIR, "pusa_v1.pt")
@@ -63,15 +62,14 @@ def ensure_model_downloaded():
         print("✅ PusaV1 downloaded.")
 
     if not os.path.exists(WAN_MODEL_PATH):
-        print("Downloading Wan-AI/Wan2.1-T2V-14B to ./model_zoo/PusaV1 ...")
+        print("Downloading Wan-AI/Wan2.1-T2V-14B to ./model_zoo/PusaV1/Wan2.1-T2V-14B ...")
         snapshot_download(
             repo_id="Wan-AI/Wan2.1-T2V-14B",
-            local_dir=WAN_DIR,
+            local_dir=WAN_MODEL_PATH,  # Changed from WAN_DIR to WAN_MODEL_PATH
             repo_type="model",
             local_dir_use_symlinks=False,
         )
         print("✅ Wan2.1-T2V-14B downloaded.")
-
 # Subclass ModelManager to force WanModelPusa