Update app_g.py

app_g.py

@@ -20,9 +20,10 @@ from huggingface_hub import snapshot_download
 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
 PUSA_ROOT = os.path.join(BASE_DIR, "PusaV1")
 MODEL_ZOO_DIR = os.path.join(PUSA_ROOT, "model_zoo")
+MODEL_ZOO_SUB_DIR = os.path.join(MODEL_ZOO_DIR , "PusaV1")
 WAN_SUBFOLDER = "Wan2.1-T2V-14B"
-WAN_MODEL_PATH = os.path.join(MODEL_ZOO_DIR, WAN_SUBFOLDER)
-LORA_PATH = os.path.join(MODEL_ZOO_DIR, "pusa_v1.pt")
+WAN_MODEL_PATH = os.path.join(MODEL_ZOO_SUB_DIR, WAN_SUBFOLDER)
+LORA_PATH = os.path.join(MODEL_ZOO_SUB_DIR, "pusa_v1.pt")
 
 # Add PUSA_ROOT to sys.path so Python can import diffsynth
 if PUSA_ROOT not in sys.path:
@@ -73,10 +74,10 @@ def ensure_model_downloaded():
         print("✅ LoRA weights (pusa_v1.pt) found.")
 
     # === List contents of model_zoo for verification
-    print(f"\n📁 Verifying files under: {MODEL_ZOO_DIR}\n")
-    for root, dirs, files in os.walk(MODEL_ZOO_DIR):
+    print(f"\n📁 Verifying files under: {MODEL_ZOO_SUB_DIR}\n")
+    for root, dirs, files in os.walk(MODEL_ZOO_SUB_DIR):
         for file in files:
-            full_path = os.path.relpath(os.path.join(root, file), start=MODEL_ZOO_DIR)
+            full_path = os.path.relpath(os.path.join(root, file), start=MODEL_ZOO_SUB_DIR)
             print(" -", full_path)