correct usage of the dropdown
Browse files
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",
|
51 |
-
gr.Checkbox(label="Enable Alpha Matting",
|
52 |
-
gr.Checkbox(label="Post-Process Mask",
|
53 |
-
gr.Checkbox(label="Only Return Mask",
|
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"),
|