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')

gr.Interface(fn=remove_background, inputs=inputs, outputs=outputs).launch(share=True)