import gradio as gr
from rembg import remove
from PIL import Image
import io

def remove_bg(image):
    image = remove(image)
    return image

iface = gr.Interface(
    fn=remove_bg,
    inputs=gr.Image(type="pil"),
    outputs=gr.Image(type="pil"),
    title="Instant Background Remover"
)

iface.launch()