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")
interface = gr.Interface(
    fn=remove_background,
    inputs=inputs,
    outputs=outputs,
    title="Remove Background",
    description="This App removes the background from an image",
    examples=[
        "examples/input/1.jpeg",
        "examples/input/2.jpeg",
        "examples/input/3.jpeg",
    ],
    cache_examples=True,
)
interface.launch(enable_queue=True)