Update app.py

app.py

@@ -1,13 +1,14 @@
 import gradio as gr
 import os
 import cv2
+from rembg import new_session, remove
+from PIL import Image
+from io import BytesIO
 
 def inference(file, mask, model, alpha_influence, segmentation_strength):
     im = cv2.imread(file, cv2.IMREAD_COLOR)
     cv2.imwrite(os.path.join("input.png"), im)
 
-    from rembg import new_session, remove
-
     input_path = 'input.png'
     output_path = 'output.png'
 
@@ -16,16 +17,19 @@ def inference(file, mask, model, alpha_influence, segmentation_strength):
             input = i.read()
             output = remove(
                 input, 
-                session=new_session(model), 
                 only_mask=(True if mask == "Mask only" else False),
-                alpha=alpha_influence,  # Control de influencia del canal alfa
-                bg_color=(0, 0, 0, segmentation_strength)  # Control de fuerza de segmentación
+                alpha_matting=True,  # Habilitar el modo alpha matting
+                alpha_matting_foreground_threshold=alpha_influence,  # Control de influencia del canal alfa
+                alpha_matting_background_threshold=1 - alpha_influence,  # Control del canal alfa para el fondo
+                alpha_matting_erode_size=int(segmentation_strength * 10),  # Control de fuerza de segmentación
+                session=new_session(model)
             )
 
             o.write(output)
-    return os.path.join("output.png")
 
-title = "RemBG_ Super"
+    return Image.open(BytesIO(output))
+
+title = "RemBG"
 description = "Gradio demo for RemBG. To use it, simply upload your image and adjust the alpha influence and segmentation strength."
 article = "<p style='text-align: center;'><a href='https://github.com/danielgatis/rembg' target='_blank'>Github Repo</a></p>"
 
@@ -59,7 +63,7 @@ gr.Interface(
         gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.5, label="Alpha Influence"),
         gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.5, label="Segmentation Strength"),
     ], 
-    gr.outputs.Image(type="filepath", label="Output"),
+    gr.outputs.Image(type="PIL", label="Output"),
     title=title,
     description=description,
     article=article,
@@ -67,4 +71,3 @@ gr.Interface(
     enable_queue=True
 ).launch()
 
-