Update app.py

app.py

@@ -1,8 +1,12 @@
 import gradio as gr
 from SegCloth import segment_clothing
 
-def segment(img):
-    return segment_clothing(img)
+def segment(img, clothes):
+    return segment_clothing(img, clothes)
 
-iface = gr.Interface(fn=segment, inputs=gr.Image(type='pil'), outputs=gr.Image())
+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()