| import gradio as gr | |
| import cv2 | |
| import numpy as np | |
| from PIL import Image | |
| from transparent_background import Remover | |
| # Load model | |
| remover = Remover(mode='fast') # custom setting | |
| # Usage for image | |
| def doo(image): | |
| img = Image.fromarray(image).convert('RGB') # read image | |
| out = remover.process(img) # default setting - transparent background | |
| out.save('output.png') # save result | |
| return out | |
| iface = gr.Interface(fn=doo, inputs="image", outputs="image") | |
| iface.launch() |