import gradio as gr
from rembg import remove

def segment(image):
     return remove(image)
     
title = "Vanish Background"
description = "Remove background for any image , To use it, simply upload your image and wait. Read more at the link below of official documentation."
article = "<p style='text-align: center;'><a href='https://github.com/danielgatis/rembg' target='_blank'>Github Repo</a></p>"
    
gr.Interface(fn=segment, inputs="image", outputs="image").launch()