Hugging Face's logo

Spaces:
ZhengPeng7
/
BiRefNet_demo
Running on Zero

App Files Community
8
ZhengPeng7 commited on
Commit
e4862f5
•
1 Parent(s): 4bb8a82

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ class ImagePreprocessor():
44
 
45
  usage_to_weights_file = {
46
  'General': 'BiRefNet',
47
- 'General-Lite': 'BiRefNet_T',
48
  'Portrait': 'BiRefNet-portrait',
49
  'DIS': 'BiRefNet-DIS5K',
50
  'HRSOD': 'BiRefNet-HRSOD',
 
44
 
45
  usage_to_weights_file = {
46
  'General': 'BiRefNet',
47
+ 'General-Lite': 'BiRefNet_lite',
48
  'Portrait': 'BiRefNet-portrait',
49
  'DIS': 'BiRefNet-DIS5K',
50
  'HRSOD': 'BiRefNet-HRSOD',