Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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('
|
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()
|