Update app.py

app.py

@@ -15,7 +15,21 @@ num_frames, width, height = 49, 832, 480
 gpu_id = 0
 device = f'cuda:{gpu_id}' if torch.cuda.is_available() else 'cpu'
 
-os.system("wget https://huggingface.co/alibaba-pai/Wan2.1-Fun-1.3B-Control/resolve/main/Wan2.1_VAE.pth")
+from huggingface_hub import snapshot_download, hf_hub_download
+snapshot_download(
+    repo_id="briaai/RMBG-2.0",
+    local_dir="ckpt/RMBG-2.0",
+    local_dir_use_symlinks=False,
+    resume_download=True,
+    repo_type="model" 
+)
+snapshot_download( # 下载整个仓库
+    repo_id="alibaba-pai/Wan2.1-Fun-1.3B-Control",
+    local_dir="ckpt/Wan2.1-Fun-1.3B-Control",
+    local_dir_use_symlinks=False,
+    resume_download=True,
+    repo_type="model"
+)
 
 rmbg_model = AutoModelForImageSegmentation.from_pretrained('briaai/RMBG-2.0', trust_remote_code=True)
 torch.set_float32_matmul_precision(['high', 'highest'][0])