|
import gradio as gr |
|
from rembg import remove |
|
|
|
title = "Remove Background" |
|
description = """ |
|
In this space, you can quickly remove background from images. |
|
|
|
⚠️ Make sure your image is of good quality and has a clear background! |
|
|
|
⚠️ Sometimes the background is not removed correctly, this is normal, because this is just a demo version! |
|
""" |
|
|
|
def segment(image): |
|
return remove(image) |
|
|
|
demo = gr.Interface(fn=segment, inputs = gr.Image(label="Input Image", interactive=True), outputs = gr.Image(label="Result Image"), title=title, description=description) |
|
demo.launch(show_api=False) |