tonyassi commited on
Commit
0c6cc50
1 Parent(s): 88157ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -2
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()