| import gradio as gr | |
| from rembg import remove | |
| from PIL import Image | |
| def remove_background(input_image): | |
| output_image = remove(input_image) | |
| return output_image | |
| inputs = gr.inputs.Image() | |
| outputs = gr.outputs.Image(type='pil') | |
| examples = [{'input': 'examples/1.jpeg'}] | |
| gr.Interface(fn=remove_background, inputs=inputs, outputs=outputs, examples=examples).launch() |