Update app.py

app.py

@@ -16,11 +16,11 @@ gpu_id = 0
 device = f'cuda:{gpu_id}' if torch.cuda.is_available() else 'cpu'
 
 from modelscope import snapshot_download
-model_dir = snapshot_download(
+model_dir = snapshot_download( # https://www.modelscope.cn/models/AI-ModelScope/RMBG-2.0
     model_id = 'AI-ModelScope/RMBG-2.0',
-    local_dir = 'ckpt/RMBG-2.0',
+    local_dir = 'test_res/RMBG-2.0',
+    ignore_file_pattern = ['onnx*'],
 )
-print(model_dir)
 
 # from huggingface_hub import snapshot_download, hf_hub_download
 # snapshot_download( # 下载整个仓库; 下briaai/RMBG-2.0需要token