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. Email me for business inquiries. """, examples=[['./1.jpg']], theme = gr.themes.Base(primary_hue="teal",secondary_hue="teal",neutral_hue="slate")) iface.launch()