Fly-ShuAI commited on
Commit
c94f775
·
verified ·
1 Parent(s): 649088b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ num_frames, width, height = 49, 832, 480
15
  gpu_id = 0
16
  device = f'cuda:{gpu_id}'
17
 
18
- rmbg_model = AutoModelForImageSegmentation.from_pretrained('ckpt/RMBG-2.0', trust_remote_code=True)
19
  torch.set_float32_matmul_precision(['high', 'highest'][0])
20
  rmbg_model.to(device)
21
  rmbg_model.eval()
 
15
  gpu_id = 0
16
  device = f'cuda:{gpu_id}'
17
 
18
+ rmbg_model = AutoModelForImageSegmentation.from_pretrained('briaai/RMBG-2.0', trust_remote_code=True)
19
  torch.set_float32_matmul_precision(['high', 'highest'][0])
20
  rmbg_model.to(device)
21
  rmbg_model.eval()