Update app.py

app.py

@@ -2,31 +2,28 @@ import os
 import sys
 import subprocess
 import time
+from huggingface_hub import snapshot_download
 
 MODEL_REPO = "tencent/HunyuanVideo-Avatar"
-BASE_DIR = os.getcwd()  # current working directory, absolute path
+BASE_DIR = os.getcwd()  # current working directory absolute path
 WEIGHTS_DIR = os.path.join(BASE_DIR, "weights")
 
-# 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 weights directory...")
+    print("⬇️ Model not found. Downloading with snapshot_download into weights directory...")
     os.makedirs(WEIGHTS_DIR, exist_ok=True)
 
-    # 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)
+    snapshot_download(
+        repo_id=MODEL_REPO,
+        local_dir=WEIGHTS_DIR,
+        local_dir_use_symlinks=False  # Force copy instead of symlink for clarity
+    )
 
     if not os.path.isfile(CHECKPOINT_FILE):
         print(f"❌ Checkpoint file not found at {CHECKPOINT_FILE} after download.")
         sys.exit(1)
+
     print("✅ Model downloaded successfully.")
 
 def run_flask_audio():
@@ -63,7 +60,7 @@ def main():
         download_model()
 
     flask_proc = run_flask_audio()
-    time.sleep(5)  # Wait a bit for flask_audio.py to start
+    time.sleep(5)  # Wait a little for flask_audio.py to start
     gradio_proc = run_gradio_ui()
 
 if __name__ == "__main__":