Hugging Face's logo

Spaces:
ghostsInTheMachine
/
BiRefNet_demo
Running on Zero

App Files Community
ZhengPeng7 commited on
Commit
bcfa392
•
1 Parent(s): 1592dab

Add BiRefNet-matting option.

Browse files
Files changed (2) hide show
  1. app.py +2 -1
  2. app_local.py +2 -1
app.py CHANGED
@@ -75,6 +75,7 @@ usage_to_weights_file = {
75
  'General': 'BiRefNet',
76
  'General-Lite': 'BiRefNet_lite',
77
  'General-Lite-2K': 'BiRefNet_lite-2K',
 
78
  'Portrait': 'BiRefNet-portrait',
79
  'DIS': 'BiRefNet-DIS5K',
80
  'HRSOD': 'BiRefNet-HRSOD',
@@ -220,7 +221,7 @@ tab_batch = gr.Interface(
220
  demo = gr.TabbedInterface(
221
  [tab_image, tab_text, tab_batch],
222
  ['image', 'text', 'batch'],
223
- title="BiRefNet demo for subject extraction (general / salient / camouflaged / portrait).",
224
  )
225
 
226
  if __name__ == "__main__":
 
75
  'General': 'BiRefNet',
76
  'General-Lite': 'BiRefNet_lite',
77
  'General-Lite-2K': 'BiRefNet_lite-2K',
78
+ 'Matting': 'BiRefNet-matting',
79
  'Portrait': 'BiRefNet-portrait',
80
  'DIS': 'BiRefNet-DIS5K',
81
  'HRSOD': 'BiRefNet-HRSOD',
 
221
  demo = gr.TabbedInterface(
222
  [tab_image, tab_text, tab_batch],
223
  ['image', 'text', 'batch'],
224
+ title="BiRefNet demo for subject extraction (general / matting / salient / camouflaged / portrait).",
225
  )
226
 
227
  if __name__ == "__main__":
app_local.py CHANGED
@@ -75,6 +75,7 @@ usage_to_weights_file = {
75
  'General': 'BiRefNet',
76
  'General-Lite': 'BiRefNet_lite',
77
  'General-Lite-2K': 'BiRefNet_lite-2K',
 
78
  'Portrait': 'BiRefNet-portrait',
79
  'DIS': 'BiRefNet-DIS5K',
80
  'HRSOD': 'BiRefNet-HRSOD',
@@ -220,7 +221,7 @@ tab_batch = gr.Interface(
220
  demo = gr.TabbedInterface(
221
  [tab_image, tab_text, tab_batch],
222
  ['image', 'text', 'batch'],
223
- title="BiRefNet demo for subject extraction (general / salient / camouflaged / portrait).",
224
  )
225
 
226
  if __name__ == "__main__":
 
75
  'General': 'BiRefNet',
76
  'General-Lite': 'BiRefNet_lite',
77
  'General-Lite-2K': 'BiRefNet_lite-2K',
78
+ 'Matting': 'BiRefNet-matting',
79
  'Portrait': 'BiRefNet-portrait',
80
  'DIS': 'BiRefNet-DIS5K',
81
  'HRSOD': 'BiRefNet-HRSOD',
 
221
  demo = gr.TabbedInterface(
222
  [tab_image, tab_text, tab_batch],
223
  ['image', 'text', 'batch'],
224
+ title="BiRefNet demo for subject extraction (general / matting / salient / camouflaged / portrait).",
225
  )
226
 
227
  if __name__ == "__main__":