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"),
|