Spaces:

Kunbyte
/

Lumen

Running on Zero

Fly-ShuAI commited on Jun 24

Commit

34012f4

verified ·

1 Parent(s): 006be6e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ device = f'cuda:{gpu_id}' if torch.cuda.is_available() else 'cpu'
 from modelscope import 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 = 'test_res/RMBG-2.0',
     ignore_file_pattern = ['onnx*'],
 )

 from modelscope import 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',
     ignore_file_pattern = ['onnx*'],
 )