Update app.py

app.py

@@ -1,18 +1,26 @@
 import gradio as gr
 from gradio_imageslider import ImageSlider
-from PIL import ImageFilter
-
+from loadimg import load_img
+from transformers import pipeline
+pipe = pipeline("image-segmentation", model="briaai/RMBG-1.4", trust_remote_code=True)
 
 def fn(im):
     if not im or not im[0]:
         return im
-    return (im[0], im[0].filter(filter=ImageFilter.GaussianBlur(radius=10)))
+    # avoid problems when working with images that are actually RGBA
+    im = im[0].convert('RGB') 
+    path = load_img(im,output_type="str")
+    rmbg = pipe(path)
+    # rmbg_path = load_img(rmbg,output_type="str")
+    out = (im,rmbg)
+    return out
 
 
 with gr.Blocks() as demo:
+    gr.Markdown("# RMBG1.4 tutorial")
     with gr.Group():
-        img1 = ImageSlider(label="Blur image", type="pil")
+        img1 = ImageSlider(label="RMBG", type="pil")
         img1.upload(fn, inputs=img1, outputs=img1)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(debug=True)