rdjarbeng commited on
Commit
9e90c80
·
1 Parent(s): fac2e01

added extra details for the dropdown

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -71,10 +71,10 @@ iface = gr.Interface(
71
  inputs=[
72
  gr.Image(type="pil"),
73
  gr.ColorPicker(label="Background Color", value=None), # Background color picker
74
- gr.Dropdown(choices=["", "u2net (default)", "isnet-general-use", "unet"], label="Model Selection", value=""),
75
  gr.Checkbox(label="Enable Alpha Matting", value=False),
76
- gr.Checkbox(label="Post-Process Mask", value=False),
77
- gr.Checkbox(label="Only Return Mask", value=False)
78
  ],
79
  outputs=[
80
  gr.Image(type="pil", label="Output Image"),
 
71
  inputs=[
72
  gr.Image(type="pil"),
73
  gr.ColorPicker(label="Background Color", value=None), # Background color picker
74
+ gr.Dropdown(choices=["", "u2net (default)| A pre-trained model for general use cases", "isnet-general-use |new pre-trained model for general use cases ", "unet | lightweight version of u2net model"], label="Model Selection", value=""),
75
  gr.Checkbox(label="Enable Alpha Matting", value=False),
76
+ gr.Checkbox(label="Post-Process Mask (to post process the mask to get better results)", value=False),
77
+ gr.Checkbox(label="Only Return Mask ", value=False)
78
  ],
79
  outputs=[
80
  gr.Image(type="pil", label="Output Image"),