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

def remove_bg(input_image):
    if input_image is None:
        return None
    return remove(input_image)

demo = gr.Interface(
    fn=remove_bg, 
    inputs=gr.Image(type="pil"), 
    outputs=gr.Image(type="pil")
)

if __name__ == "__main__":
    demo.launch()