import gradio as gr from SegCloth import segment_clothing def segment(img): return segment_clothing(img) iface = gr.Interface(fn=segment, inputs=gr.Image(type='pil'), outputs=gr.Image()) iface.launch()