ZhengPeng7 commited on
Commit
e7c2780
1 Parent(s): a0ef2a3

Add default value of radio.

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -37,7 +37,7 @@ class ImagePreprocessor():
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,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,