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: tony.assi.media@gmail.com">Email me</a> for business inquiries.
|
20 |
+
""",
|
21 |
+
theme = gr.themes.Base(primary_hue="teal",secondary_hue="teal",neutral_hue="slate"))
|
22 |
iface.launch()
|