Spaces:
Running
on
Zero
Running
on
Zero
bluestyle97
commited on
Update freesplatter/webui/runner.py
Browse files
freesplatter/webui/runner.py
CHANGED
@@ -75,7 +75,8 @@ class FreeSplatterRunner:
|
|
75 |
|
76 |
# background remover
|
77 |
self.rembg = AutoModelForImageSegmentation.from_pretrained(
|
78 |
-
"ZhengPeng7/BiRefNet",
|
|
|
79 |
trust_remote_code=True,
|
80 |
).to(device)
|
81 |
self.rembg.eval()
|
|
|
75 |
|
76 |
# background remover
|
77 |
self.rembg = AutoModelForImageSegmentation.from_pretrained(
|
78 |
+
# "ZhengPeng7/BiRefNet",
|
79 |
+
"briaai/RMBG-2.0",
|
80 |
trust_remote_code=True,
|
81 |
).to(device)
|
82 |
self.rembg.eval()
|