| 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() |