Hugging Face's logo

Spaces:
randomtable
/
BiRefNet_Backgroun_Removal
Running

App Files Community
ZhengPeng7 commited on
Commit
0891619
·
verified ·
1 Parent(s): 04086f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ class ImagePreprocessor():
36
 
37
 
38
  from transformers import AutoModelForImageSegmentation
39
- birefnet = AutoModelForImageSegmentation.from_pretrained('zhengpeng7/BiRefNet-portrait', trust_remote_code=True)
40
  birefnet.to(device)
41
  birefnet.eval()
42
 
 
36
 
37
 
38
  from transformers import AutoModelForImageSegmentation
39
+ birefnet = AutoModelForImageSegmentation.from_pretrained('zhengpeng7/BiRefNet', trust_remote_code=True)
40
  birefnet.to(device)
41
  birefnet.eval()
42