rdjarbeng commited on
Commit
09cdf11
·
1 Parent(s): 5c0ed2c

correct usage of the dropdown

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -1,7 +1,6 @@
1
  import gradio as gr
2
  from rembg import remove, new_session
3
  from PIL import Image
4
- import numpy as np
5
  import logging
6
 
7
  # Set up logging
@@ -47,10 +46,10 @@ iface = gr.Interface(
47
  inputs=[
48
  gr.Image(type="pil"),
49
  gr.ColorPicker(label="Background Color", value=None), # Background color picker
50
- gr.Dropdown(choices=["u2net", "isnet-general-use", "unet"], label="Model Selection", default="u2net"),
51
- gr.Checkbox(label="Enable Alpha Matting", default=False),
52
- gr.Checkbox(label="Post-Process Mask", default=False),
53
- gr.Checkbox(label="Only Return Mask", default=False)
54
  ],
55
  outputs=[
56
  gr.Image(type="pil", label="Output Image"),
 
1
  import gradio as gr
2
  from rembg import remove, new_session
3
  from PIL import Image
 
4
  import logging
5
 
6
  # Set up logging
 
46
  inputs=[
47
  gr.Image(type="pil"),
48
  gr.ColorPicker(label="Background Color", value=None), # Background color picker
49
+ gr.Dropdown(choices=["u2net", "isnet-general-use", "unet"], label="Model Selection", value="u2net"),
50
+ gr.Checkbox(label="Enable Alpha Matting", value=False),
51
+ gr.Checkbox(label="Post-Process Mask", value=False),
52
+ gr.Checkbox(label="Only Return Mask", value=False)
53
  ],
54
  outputs=[
55
  gr.Image(type="pil", label="Output Image"),