Update app.py

app.py

@@ -31,10 +31,10 @@ snapshot_download( # 下载整个仓库
     repo_type="model"
 )
 
-rmbg_model = AutoModelForImageSegmentation.from_pretrained('briaai/RMBG-2.0', trust_remote_code=True)
-torch.set_float32_matmul_precision(['high', 'highest'][0])
-rmbg_model.to(device)
-rmbg_model.eval()
+# rmbg_model = AutoModelForImageSegmentation.from_pretrained('ckpt/RMBG-2.0', trust_remote_code=True)
+# torch.set_float32_matmul_precision(['high', 'highest'][0])
+# rmbg_model.to(device)
+# rmbg_model.eval()
 
 model_manager = ModelManager(device="cpu") # 1.3b: device=cpu: uses 6G VRAM, device=device: uses 16G VRAM; about 1-2 min per video