File size: 1,471 Bytes
bddf3b8
 
 
88157ff
 
bddf3b8
88157ff
34e47b3
88157ff
0c6cc50
34e47b3
 
 
0c6cc50
 
 
 
 
 
 
 
1fae26d
0c6cc50
bddf3b8
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
import gradio as gr
from SegCloth import segment_clothing

def segment(img, clothes):
    return segment_clothing(img, clothes)

iface = gr.Interface(fn=segment, 
                     inputs=[gr.Image(type='pil', label='Image'),
                             gr.Dropdown(choices=["Hat", "Upper-clothes", "Skirt", "Pants", "Dress", "Belt", "Left-shoe", "Right-shoe", "Scarf"],
                                         value=["Hat", "Upper-clothes", "Skirt", "Pants", "Dress", "Belt", "Left-shoe", "Right-shoe", "Scarf"],
                                         multiselect=True,
                                         label='Clothing')], 
                     outputs=gr.Image(label='Clothing Segmentation'),
                     title='Clothing Segmentation',
                     description="""
                     by [Tony Assi](https://www.tonyassi.com/)
                     
                     This space uses the [Clothing Segmentation Github Repo](https://github.com/TonyAssi/Segment-Clothing). Please ❀️ this Space.
        
                     I build custom AI apps for companies. <a href="mailto: tony.assi.media@gmail.com">Email me</a> for business inquiries.
                     """,
                     examples=[['./1.jpg', ["Hat", "Upper-clothes", "Skirt", "Pants", "Dress", "Belt", "Left-shoe", "Right-shoe", "Scarf"]]],
                     theme = gr.themes.Base(primary_hue="teal",secondary_hue="teal",neutral_hue="slate"))
iface.launch()