Update app_g.py

app_g.py

@@ -12,7 +12,7 @@ from huggingface_hub import snapshot_download
 
 # === Setup Paths ===
 PUSA_ROOT = os.path.abspath("./PusaV1")
-MODEL_ZOO_DIR = os.path.join(PUSA_ROOT, "modelzoo")
+MODEL_ZOO_DIR = os.path.join(PUSA_ROOT, "model_zoo")
 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")
@@ -32,7 +32,7 @@ if not os.path.exists(DIFFSYNTH_PATH):
 # === Ensure model and weights are downloaded ===
 def ensure_model_downloaded():
     if not os.path.exists(MODEL_ZOO_DIR):
-        print("Downloading RaphaelLiu/PusaV1 to ./PusaV1/modelzoo ...")
+        print("Downloading RaphaelLiu/PusaV1 to ./PusaV1/model_zoo ...")
         snapshot_download(
             repo_id="RaphaelLiu/PusaV1",
             local_dir=MODEL_ZOO_DIR,
@@ -42,9 +42,9 @@ def ensure_model_downloaded():
         print("✅ PusaV1 base model downloaded.")
 
     if not os.path.exists(WAN_MODEL_PATH):
-        print("Downloading Wan-AI/Wan2.1-T2V-14B to ./PusaV1/modelzoo/Wan2.1-T2V-14B ...")
+        print("Downloading Wan-AI/Wan2.1-T2V-14B to ./PusaV1/model_zoo/Wan2.1-T2V-14B ...")
         snapshot_download(
-            repo_id="Wan-AI/Wan2.1-T2V-14B-Diffusers",  # Adjust if necessary
+            repo_id="Wan-AI/Wan2.1-T2V-14B-Diffusers",
             local_dir=WAN_MODEL_PATH,
             repo_type="model",
             local_dir_use_symlinks=False,
@@ -57,6 +57,14 @@ def ensure_model_downloaded():
             f"Please ensure the file exists or is downloaded manually."
         )
 
+    # ✅ Add this to list all files in modelzoo
+    print(f"\n📁 Verifying files under: {MODEL_ZOO_DIR}\n")
+    for root, dirs, files in os.walk(MODEL_ZOO_DIR):
+        for file in files:
+            full_path = os.path.relpath(os.path.join(root, file), start=MODEL_ZOO_DIR)
+            print(" -", full_path)
+
+
 
 
 import gradio as gr
@@ -83,7 +91,8 @@ class PusaVideoDemo:
         self.multi_frames_pipe = None
         self.v2v_pipe = None
         self.t2v_pipe = None
-        self.base_dir = "PusaV1/model_zoo/PusaV1/Wan2.1-T2V-14B"
+        self.base_dir = os.path.join("PusaV1", "model_zoo", "Wan2.1-T2V-14B")
+
         self.output_dir = "outputs"
         os.makedirs(self.output_dir, exist_ok=True)