| 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'), | |
| gr.Dropdown(choices=["Hat", "Upper-clothes", "Skirt", "Pants", "Dress", "Belt", "Left-shoe", "Right-shoe", "Scarf"], | |
| multiselect=True)], | |
| outputs=gr.Image()) | |
| iface.launch() |