Update app.py
Browse files
app.py
CHANGED
|
@@ -32,7 +32,12 @@ def remove_background(img):
|
|
| 32 |
|
| 33 |
return segformer_b2_clothes_text, segformer_b2_clothes_result, rembg_text, rembg_result
|
| 34 |
|
| 35 |
-
iface = gr.Interface(fn=remove_background,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 36 |
inputs=gr.Image(type='pil'),
|
| 37 |
outputs=[gr.Markdown(),
|
| 38 |
gr.Image(label='segformer_b2_clothes', type='pil'),
|
|
|
|
| 32 |
|
| 33 |
return segformer_b2_clothes_text, segformer_b2_clothes_result, rembg_text, rembg_result
|
| 34 |
|
| 35 |
+
iface = gr.Interface(fn=remove_background,
|
| 36 |
+
title='Remove Background Comparison',
|
| 37 |
+
description="""
|
| 38 |
+
Compares [mattmdjaga/segformer_b2_clothes](https://huggingface.co/mattmdjaga/segformer_b2_clothes) and [rembg](https://huggingface.co/spaces/openskyml/remove-background-on-image) background removal.
|
| 39 |
+
""",
|
| 40 |
+
theme = gr.themes.Base(primary_hue="teal",secondary_hue="teal",neutral_hue="slate"),
|
| 41 |
inputs=gr.Image(type='pil'),
|
| 42 |
outputs=[gr.Markdown(),
|
| 43 |
gr.Image(label='segformer_b2_clothes', type='pil'),
|