| import gradio as gr | |
| import os | |
| os.system("pip install rembg") | |
| os.system("pip install opencv-python") | |
| import rembg | |
| 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() |