rembg / app.py
hysts's picture
hysts HF Staff
Update
b15d723
raw
history blame contribute delete
966 Bytes
#!/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)