Update app.py
Browse files
app.py
CHANGED
|
@@ -5,8 +5,18 @@ def segment(img, clothes):
|
|
| 5 |
return segment_clothing(img, clothes)
|
| 6 |
|
| 7 |
iface = gr.Interface(fn=segment,
|
| 8 |
-
inputs=[gr.Image(type='pil'),
|
| 9 |
gr.Dropdown(choices=["Hat", "Upper-clothes", "Skirt", "Pants", "Dress", "Belt", "Left-shoe", "Right-shoe", "Scarf"],
|
|
|
|
| 10 |
multiselect=True)],
|
| 11 |
-
outputs=gr.Image()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
iface.launch()
|
|
|
|
| 5 |
return segment_clothing(img, clothes)
|
| 6 |
|
| 7 |
iface = gr.Interface(fn=segment,
|
| 8 |
+
inputs=[gr.Image(type='pil'),
|
| 9 |
gr.Dropdown(choices=["Hat", "Upper-clothes", "Skirt", "Pants", "Dress", "Belt", "Left-shoe", "Right-shoe", "Scarf"],
|
| 10 |
+
value=["Hat", "Upper-clothes", "Skirt", "Pants", "Dress", "Belt", "Left-shoe", "Right-shoe", "Scarf"],
|
| 11 |
multiselect=True)],
|
| 12 |
+
outputs=gr.Image(),
|
| 13 |
+
title='Clothing Segmentation',
|
| 14 |
+
description="""
|
| 15 |
+
by [Tony Assi](https://www.tonyassi.com/)
|
| 16 |
+
|
| 17 |
+
This space uses the [Clothing Segmentation Github Repo](https://github.com/TonyAssi/Segment-Clothing). Please ❤️ this Space.
|
| 18 |
+
|
| 19 |
+
I build custom AI apps for companies. <a href="mailto: [email protected]">Email me</a> for business inquiries.
|
| 20 |
+
""",
|
| 21 |
+
theme = gr.themes.Base(primary_hue="teal",secondary_hue="teal",neutral_hue="slate"))
|
| 22 |
iface.launch()
|