Update app.py

app.py

@@ -1,22 +1,27 @@
 import os
 import sys
 import subprocess
-from huggingface_hub import snapshot_download
+import time
 
-MODEL_BASE = "./weights"
-MODEL_DIR = os.path.join(MODEL_BASE, "ckpts/hunyuan-video-t2v-720p")
-CHECKPOINT_FILE = os.path.join(MODEL_DIR, "transformers/mp_rank_00_model_states.pt")
+MODEL_REPO = "tencent/HunyuanVideo-Avatar"
+BASE_DIR = "HunyuanVideo-Avatar"
+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
 
 def download_model():
-    print("⬇️  Model not found. Downloading from Hugging Face...")
-    os.makedirs(MODEL_DIR, exist_ok=True)
-    snapshot_download(
-        repo_id="tencent/HunyuanVideo-Avatar",
-        local_dir=MODEL_DIR,
-        local_dir_use_symlinks=False,
-    )
+    print("⬇️  Model not found. Downloading with huggingface-cli inside HunyuanVideo-Avatar/weights...")
+    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)
+    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("❌ Failed to download model checkpoint. Exiting.")
+        print(f"❌ Checkpoint file not found at {CHECKPOINT_FILE} after download.")
         sys.exit(1)
     print("✅ Model downloaded successfully.")
 
@@ -38,7 +43,6 @@ def run_flask_audio():
         "--use-deepcache", "1",
         "--flow-shift-eval-video", "5.0"
     ]
-    # Start process in background
     proc = subprocess.Popen(cmd)
     return proc
 
@@ -55,15 +59,8 @@ def main():
         download_model()
 
     flask_proc = run_flask_audio()
-    # Optionally wait a few seconds for flask_audio.py to start before starting Gradio UI
-    import time
-    time.sleep(5)
-
+    time.sleep(5)  # Wait for flask_audio.py to spin up
     gradio_proc = run_gradio_ui()
 
-    # Wait for both processes to exit if you want to block here, or just exit
-    # flask_proc.wait()
-    # gradio_proc.wait()
-
 if __name__ == "__main__":
     main()