Spaces:
Running
on
Zero
Running
on
Zero
ZhengPeng7
commited on
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,
|