app.py

import gradio as gr
from rembg import remove

def segment(image):
     return remove(image)
    
gr.Interface(fn=segment, inputs="image", outputs="image").launch()