Update app.py
Browse files
app.py
CHANGED
|
@@ -134,7 +134,9 @@ except Exception as e:
|
|
| 134 |
|
| 135 |
# ── RMBG 2.0 抠图模型 ──
|
| 136 |
rmbg = AutoModelForImageSegmentation.from_pretrained(
|
| 137 |
-
"briaai/RMBG-2.0",
|
|
|
|
|
|
|
| 138 |
)
|
| 139 |
rmbg.to(device)
|
| 140 |
rmbg.eval()
|
|
|
|
| 134 |
|
| 135 |
# ── RMBG 2.0 抠图模型 ──
|
| 136 |
rmbg = AutoModelForImageSegmentation.from_pretrained(
|
| 137 |
+
"briaai/RMBG-2.0",
|
| 138 |
+
trust_remote_code=True,
|
| 139 |
+
token=os.environ.get("HF_TOKEN"),
|
| 140 |
)
|
| 141 |
rmbg.to(device)
|
| 142 |
rmbg.eval()
|