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 = "Eliminar Background"

description = ""

article = ""


gr.Interface(
    inference, 
    gr.inputs.Image(type="filepath", label="Input"), 
    gr.outputs.Image(type="pil", label="Output"),
    title=title,
    description=description,
    article=article,
    css="Footer {visibility: hidden}"
    ).launch()