main.py

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