Update app.py

app.py

@@ -30,10 +30,16 @@ def download_model():
     print("⬇️  Model not found. Downloading with snapshot_download into weights directory...")
     os.makedirs(WEIGHTS_DIR, exist_ok=True)
 
+    # snapshot_download(
+    #     repo_id=MODEL_REPO,
+    #     local_dir=WEIGHTS_DIR,
+    #     local_dir_use_symlinks=False
+    # )
     snapshot_download(
-        repo_id="https://huggingface.co/tencent/HunyuanVideo-Avatar/blob/main/ckpts/hunyuan-video-t2v-720p/transformers/mp_rank_00_model_states_fp8.pt",
-        local_dir=WEIGHTS_DIR,
-        local_dir_use_symlinks=False
+    repo_id="tencent/HunyuanVideo-Avatar",  # Just the repo ID
+    local_dir=WEIGHTS_DIR,
+    local_dir_use_symlinks=False,
+    allow_patterns=["ckpts/hunyuan-video-t2v-720p/transformers/mp_rank_00_model_states_fp8.pt"]
     )
 
     if not os.path.isfile(CHECKPOINT_FILE):