import os import gradio as gr def process(im): return im title = "U-2-Net" description = "Gradio demo for U-2-Net, https://github.com/xuebinqin/U-2-Net" article = "" gr.Interface( process, [gr.inputs.Image(type="pil", label="Input") ], gr.outputs.Image(type="pil", label="Output"), title=title, description=description, article=article, examples=[], allow_flagging=False, allow_screenshot=False ).launch(enable_queue=True,cache_examples=True)