Update app.py

app.py

@@ -18,19 +18,11 @@ device = f'cuda:{gpu_id}' if torch.cuda.is_available() else 'cpu'
 from modelscope import snapshot_download
 model_dir = snapshot_download(
     model_id = 'AI-ModelScope/RMBG-2.0',
-    local_dir = 'ckpt/',
+    local_dir = 'ckpt/RMBG-2.0',
 )
 print(model_dir)
 
 # from huggingface_hub import snapshot_download, hf_hub_download
-# hf_hub_download(
-#     repo_id="alibaba-pai/Wan2.1-Fun-1.3B-Control",
-#     filename="Wan2.1_VAE.pth",
-#     local_dir="ckpt/Wan2.1-Fun-1.3B-Control/",
-#     local_dir_use_symlinks=False,
-#     resume_download=True,
-# )
-
 # snapshot_download( # 下载整个仓库; 下briaai/RMBG-2.0需要token
 #     repo_id="alibaba-pai/Wan2.1-Fun-1.3B-Control",
 #     local_dir="ckpt/Wan2.1-Fun-1.3B-Control",