Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -84,7 +84,7 @@ usage_to_weights_file = {
|
|
| 84 |
'General-legacy': 'BiRefNet-legacy'
|
| 85 |
}
|
| 86 |
|
| 87 |
-
birefnet = AutoModelForImageSegmentation.from_pretrained('zhengpeng7/BiRefNet_lite'
|
| 88 |
birefnet.to(device)
|
| 89 |
birefnet.eval()
|
| 90 |
|
|
|
|
| 84 |
'General-legacy': 'BiRefNet-legacy'
|
| 85 |
}
|
| 86 |
|
| 87 |
+
birefnet = AutoModelForImageSegmentation.from_pretrained('zhengpeng7/BiRefNet_lite', trust_remote_code=True)
|
| 88 |
birefnet.to(device)
|
| 89 |
birefnet.eval()
|
| 90 |
|