added extra details for the dropdown
Browse files
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"),
|