Update app.py

app.py

@@ -7,20 +7,23 @@ MODEL_REPO = "tencent/HunyuanVideo-Avatar"
 BASE_DIR = os.getcwd()  # current working directory, absolute path
 WEIGHTS_DIR = os.path.join(BASE_DIR, "weights")
 
-CHECKPOINT_FILE = os.path.join(WEIGHTS_DIR, "transformers/mp_rank_00_model_states.pt")  # Adjust if needed
+# This path assumes the checkpoint is inside weights/transformers/...
+CHECKPOINT_FILE = os.path.join(WEIGHTS_DIR, "transformers", "mp_rank_00_model_states.pt")
 
 def download_model():
-    print("⬇️  Model not found. Downloading with huggingface-cli inside HunyuanVideo-Avatar/weights...")
+    print("⬇️  Model not found. Downloading with huggingface-cli inside weights directory...")
     os.makedirs(WEIGHTS_DIR, exist_ok=True)
-    
-    # Run huggingface-cli download inside WEIGHTS_DIR
-    cmd = ["huggingface-cli", "download", MODEL_REPO, "--local-dir", "./"]
-    proc = subprocess.run(cmd, cwd=WEIGHTS_DIR)
+
+    # Use --local-dir as absolute path (important!)
+    cmd = [
+        "huggingface-cli", "download", MODEL_REPO,
+        "--local-dir", WEIGHTS_DIR
+    ]
+    proc = subprocess.run(cmd)
     if proc.returncode != 0:
         print("❌ huggingface-cli download failed.")
         sys.exit(1)
-    
-    # Check if checkpoint file exists now
+
     if not os.path.isfile(CHECKPOINT_FILE):
         print(f"❌ Checkpoint file not found at {CHECKPOINT_FILE} after download.")
         sys.exit(1)
@@ -60,7 +63,7 @@ def main():
         download_model()
 
     flask_proc = run_flask_audio()
-    time.sleep(5)  # Wait for flask_audio.py to spin up
+    time.sleep(5)  # Wait a bit for flask_audio.py to start
     gradio_proc = run_gradio_ui()
 
 if __name__ == "__main__":