Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
e7c2780
1
Parent(s):
a0ef2a3
Add default value of radio.
Browse files
app.py
CHANGED
|
@@ -37,7 +37,7 @@ class ImagePreprocessor():
|
|
| 37 |
|
| 38 |
usage_to_weights_file = {
|
| 39 |
'General': 'BiRefNet',
|
| 40 |
-
'General-
|
| 41 |
'Portrait': 'BiRefNet-portrait',
|
| 42 |
'DIS': 'BiRefNet-DIS5K',
|
| 43 |
'HRSOD': 'BiRefNet-HRSOD',
|
|
@@ -103,7 +103,7 @@ demo = gr.Interface(
|
|
| 103 |
inputs=[
|
| 104 |
'image',
|
| 105 |
gr.Textbox(lines=1, placeholder="Type the resolution (`WxH`) you want, e.g., `512x512`. Higher resolutions can be much slower for inference.", label="Resolution"),
|
| 106 |
-
gr.Radio(list(usage_to_weights_file.keys()), label="Weights", info="Choose the weights you want.")
|
| 107 |
],
|
| 108 |
outputs=ImageSlider(),
|
| 109 |
examples=examples,
|
|
|
|
| 37 |
|
| 38 |
usage_to_weights_file = {
|
| 39 |
'General': 'BiRefNet',
|
| 40 |
+
'General-Lite': 'BiRefNet_T',
|
| 41 |
'Portrait': 'BiRefNet-portrait',
|
| 42 |
'DIS': 'BiRefNet-DIS5K',
|
| 43 |
'HRSOD': 'BiRefNet-HRSOD',
|
|
|
|
| 103 |
inputs=[
|
| 104 |
'image',
|
| 105 |
gr.Textbox(lines=1, placeholder="Type the resolution (`WxH`) you want, e.g., `512x512`. Higher resolutions can be much slower for inference.", label="Resolution"),
|
| 106 |
+
gr.Radio(list(usage_to_weights_file.keys()), value='General', label="Weights", info="Choose the weights you want.")
|
| 107 |
],
|
| 108 |
outputs=ImageSlider(),
|
| 109 |
examples=examples,
|