tomaseo2022's picture
Update app.py
f25b978
raw
history blame
571 Bytes
import os
os.system("pip3 install opencv-python")
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 Fondo de Imagen"
description = "Elimina el fondo de cualquier imagen de forma fácil y rápida."
article = ""
gr.Interface(
inference,
gr.inputs.Image(type="filepath", label="Input"),
gr.outputs.Image(type="pil", label="Output"),
title=title,
description=description,
article=article
).launch()