Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
bcfa392
1
Parent(s):
1592dab
Add BiRefNet-matting option.
Browse files- app.py +2 -1
- 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__":
|