Spaces:

Daankular
/

Image2Model

Running on Zero

Daankular commited on 30 days ago

Commit

dfd6c0e

1 Parent(s): 30a271d

Revert to 1038lab/RMBG-2.0 (briaai is gated)

Files changed (1) hide show

app.py CHANGED Viewed

@@ -250,7 +250,7 @@ def load_triposg():
     try:
         from transformers import AutoModelForImageSegmentation
         _rmbg_net = AutoModelForImageSegmentation.from_pretrained(
-            "briaai/RMBG-2.0", trust_remote_code=True
         )
         torch.set_float32_matmul_precision("high")
         _rmbg_net.to(DEVICE)

     try:
         from transformers import AutoModelForImageSegmentation
         _rmbg_net = AutoModelForImageSegmentation.from_pretrained(
+            "1038lab/RMBG-2.0", trust_remote_code=True
         )
         torch.set_float32_matmul_precision("high")
         _rmbg_net.to(DEVICE)