ZhengPeng7 commited on
Commit
46304c9
·
verified ·
1 Parent(s): 58ab236

Correct the wrong HF model name of BiRefNet_512x512. Add the BiRefNet_lite-Anime option.

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -83,9 +83,10 @@ usage_to_weights_file = {
83
  'Matting-HR': 'BiRefNet_HR-matting',
84
  'Matting': 'BiRefNet-matting',
85
  'Portrait': 'BiRefNet-portrait',
86
- 'General-reso_512': 'BiRefNet-reso_512',
87
  'General-Lite': 'BiRefNet_lite',
88
  'General-Lite-2K': 'BiRefNet_lite-2K',
 
89
  'DIS': 'BiRefNet-DIS5K',
90
  'HRSOD': 'BiRefNet-HRSOD',
91
  'COD': 'BiRefNet-COD',
 
83
  'Matting-HR': 'BiRefNet_HR-matting',
84
  'Matting': 'BiRefNet-matting',
85
  'Portrait': 'BiRefNet-portrait',
86
+ 'General-reso_512': 'BiRefNet_512x512',
87
  'General-Lite': 'BiRefNet_lite',
88
  'General-Lite-2K': 'BiRefNet_lite-2K',
89
+ 'Anime-Lite': 'BiRefNet_lite-Anime',
90
  'DIS': 'BiRefNet-DIS5K',
91
  'HRSOD': 'BiRefNet-HRSOD',
92
  'COD': 'BiRefNet-COD',