| 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: [email protected]">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() |