Commit
·
953b294
1
Parent(s):
6a61ba8
Refactor Gradio interface to enhance input and output organization, update background choice label, and streamline button click event configuration for improved usability and clarity.
Browse files
app.py
CHANGED
|
@@ -223,47 +223,73 @@ with gr.Blocks(
|
|
| 223 |
with gr.Column():
|
| 224 |
input_image = gr.Image(
|
| 225 |
label="Image input",
|
| 226 |
-
image_mode="RGBA",
|
| 227 |
-
sources="upload",
|
| 228 |
type="pil",
|
|
|
|
| 229 |
)
|
| 230 |
bg_choice = gr.Radio(
|
| 231 |
-
["Alpha as mask", "Auto Remove background"],
|
| 232 |
value="Auto Remove background",
|
| 233 |
-
label="
|
| 234 |
)
|
| 235 |
fg_ratio = gr.Slider(
|
| 236 |
label="Foreground Ratio",
|
| 237 |
minimum=0.5,
|
| 238 |
maximum=1.0,
|
| 239 |
value=1.0,
|
| 240 |
-
step=0.05
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 241 |
)
|
| 242 |
-
bg_color = gr.ColorPicker(label="Background Color", value="#7F7F7F", interactive=False)
|
| 243 |
-
seed_val = gr.Number(value=1234, label="seed", precision=0)
|
| 244 |
-
guidance = gr.Number(value=5.5, minimum=3, maximum=10, label="guidance_scale")
|
| 245 |
-
steps = gr.Number(value=30, minimum=30, maximum=100, label="sample steps", precision=0)
|
| 246 |
generate_btn = gr.Button("Generate")
|
| 247 |
|
| 248 |
with gr.Column():
|
| 249 |
-
output_rgb = gr.Image(
|
| 250 |
-
output_ccm = gr.Image(
|
| 251 |
-
output_glb = gr.Model3D(label="Output GLB"
|
| 252 |
|
| 253 |
# Connect the button click event
|
| 254 |
generate_btn.click(
|
| 255 |
fn=generate,
|
| 256 |
-
inputs=[
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 257 |
outputs=[output_rgb, output_ccm, output_glb],
|
| 258 |
-
|
| 259 |
)
|
| 260 |
|
| 261 |
# Add examples without caching
|
| 262 |
gr.Examples(
|
| 263 |
examples=[[os.path.join("examples", i)] for i in os.listdir("examples")],
|
| 264 |
inputs=input_image,
|
| 265 |
-
cache_examples=False
|
| 266 |
)
|
| 267 |
|
| 268 |
if __name__ == "__main__":
|
| 269 |
-
demo.launch()
|
|
|
|
| 223 |
with gr.Column():
|
| 224 |
input_image = gr.Image(
|
| 225 |
label="Image input",
|
|
|
|
|
|
|
| 226 |
type="pil",
|
| 227 |
+
sources=["upload"]
|
| 228 |
)
|
| 229 |
bg_choice = gr.Radio(
|
| 230 |
+
choices=["Alpha as mask", "Auto Remove background"],
|
| 231 |
value="Auto Remove background",
|
| 232 |
+
label="Background choice"
|
| 233 |
)
|
| 234 |
fg_ratio = gr.Slider(
|
| 235 |
label="Foreground Ratio",
|
| 236 |
minimum=0.5,
|
| 237 |
maximum=1.0,
|
| 238 |
value=1.0,
|
| 239 |
+
step=0.05
|
| 240 |
+
)
|
| 241 |
+
bg_color = gr.ColorPicker(
|
| 242 |
+
label="Background Color",
|
| 243 |
+
value="#7F7F7F",
|
| 244 |
+
interactive=False
|
| 245 |
+
)
|
| 246 |
+
seed_val = gr.Number(
|
| 247 |
+
value=1234,
|
| 248 |
+
label="Seed",
|
| 249 |
+
precision=0
|
| 250 |
+
)
|
| 251 |
+
guidance = gr.Number(
|
| 252 |
+
value=5.5,
|
| 253 |
+
minimum=3,
|
| 254 |
+
maximum=10,
|
| 255 |
+
label="Guidance scale"
|
| 256 |
+
)
|
| 257 |
+
steps = gr.Number(
|
| 258 |
+
value=30,
|
| 259 |
+
minimum=30,
|
| 260 |
+
maximum=100,
|
| 261 |
+
label="Sample steps",
|
| 262 |
+
precision=0
|
| 263 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 264 |
generate_btn = gr.Button("Generate")
|
| 265 |
|
| 266 |
with gr.Column():
|
| 267 |
+
output_rgb = gr.Image(label="Output RGB image")
|
| 268 |
+
output_ccm = gr.Image(label="Output CCM image")
|
| 269 |
+
output_glb = gr.Model3D(label="Output GLB")
|
| 270 |
|
| 271 |
# Connect the button click event
|
| 272 |
generate_btn.click(
|
| 273 |
fn=generate,
|
| 274 |
+
inputs=[
|
| 275 |
+
input_image,
|
| 276 |
+
bg_choice,
|
| 277 |
+
fg_ratio,
|
| 278 |
+
bg_color,
|
| 279 |
+
seed_val,
|
| 280 |
+
guidance,
|
| 281 |
+
steps
|
| 282 |
+
],
|
| 283 |
outputs=[output_rgb, output_ccm, output_glb],
|
| 284 |
+
api_name="predict"
|
| 285 |
)
|
| 286 |
|
| 287 |
# Add examples without caching
|
| 288 |
gr.Examples(
|
| 289 |
examples=[[os.path.join("examples", i)] for i in os.listdir("examples")],
|
| 290 |
inputs=input_image,
|
| 291 |
+
cache_examples=False
|
| 292 |
)
|
| 293 |
|
| 294 |
if __name__ == "__main__":
|
| 295 |
+
demo.launch(show_api=True)
|