| import gradio as gr | |
| from gradio_imageslider import ImageSlider | |
| from loadimg import load_img | |
| from transformers import pipeline | |
| pipe = pipeline("image-segmentation", model="briaai/RMBG-1.4", trust_remote_code=True) | |
| def fn(image): | |
| im = load_img(image) | |
| im = im.convert('RGB') | |
| path = load_img(im,output_type="str") | |
| rmbg = pipe(path) | |
| out = (rmbg , im) | |
| return out | |
| slider1 = ImageSlider(label="RMBG", type="pil") | |
| slider2 = ImageSlider(label="RMBG", type="pil") | |
| image = gr.Image(label="Upload an image") | |
| text = gr.Textbox(label="Paste an image URL") | |
| tab1 = gr.Interface(fn,inputs= image, outputs= slider1, api_name="image") | |
| tab2 = gr.Interface(fn,inputs= text, outputs= slider2, api_name="text") | |
| demo = gr.TabbedInterface([tab1,tab2],["image","text"],title="RMBG with image slider") | |
| if __name__ == "__main__": | |
| demo.launch() |