import os
os.system("/usr/local/bin/python -m pip install --upgrade pip")
import gradio as gr
from rembg import remove
import cv2

def inference(img):
    input_img = cv2.imread(img)
    output = remove(input_img[:, :, [2,1,0]])
    return output

title = "Remove Background App"

description = "Upload images and click submit"

gr.Interface(
    inference, 
    gr.inputs.Image(type="filepath", label="Input"), 
    gr.outputs.Image(type="pil", label="Output"),
    ).launch()