#!/usr/bin/env python import gradio as gr import PIL.Image import rembg DESCRIPTION = "# rembg - Image background removal" def remove_background(image: PIL.Image.Image) -> tuple[PIL.Image.Image, PIL.Image.Image]: """Remove background from image using `rembg`. Args: image: The image to remove the background from. Returns: A tuple containing the image with the background removed and the original image. """ return rembg.remove(image), image with gr.Blocks(css_paths="style.css") as demo: gr.Markdown(DESCRIPTION) with gr.Row(): with gr.Column(): image = gr.Image(type="pil") run_button = gr.Button() with gr.Column(): out = gr.ImageSlider() gr.Examples(examples=["cats.jpg"], fn=remove_background, inputs=image, outputs=out) run_button.click(fn=remove_background, inputs=image, outputs=out) if __name__ == "__main__": demo.launch(mcp_server=True)