bluestyle97 commited on
Commit
944d7dc
·
verified ·
1 Parent(s): 1d262d9

Update freesplatter/webui/runner.py

Browse files
Files changed (1) hide show
  1. freesplatter/webui/runner.py +2 -1
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()