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()
|