Spaces:

TencentARC
/

FreeSplatter

Runtime error

bluestyle97 commited on 6 days ago

Commit

6f3d139

verified ·

1 Parent(s): 2717ed4

switch to birefnet

Files changed (1) hide show

freesplatter/webui/runner.py CHANGED Viewed

@@ -75,8 +75,8 @@ class FreeSplatterRunner:
         # background remover
         self.rembg = AutoModelForImageSegmentation.from_pretrained(
-            # "ZhengPeng7/BiRefNet",
-            "briaai/RMBG-2.0",
             trust_remote_code=True,
         ).to(device)
         self.rembg.eval()

         # background remover
         self.rembg = AutoModelForImageSegmentation.from_pretrained(
+            "ZhengPeng7/BiRefNet",
+            # "briaai/RMBG-2.0",
             trust_remote_code=True,
         ).to(device)
         self.rembg.eval()