Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import os
 import cv2
 
-def inference(file, mask, model):
+def inference(file, mask, model, alpha_influence, segmentation_strength):
     im = cv2.imread(file, cv2.IMREAD_COLOR)
     cv2.imwrite(os.path.join("input.png"), im)
 
@@ -16,20 +16,18 @@ def inference(file, mask, model):
             input = i.read()
             output = remove(
                 input, 
-                session = new_session(model), 
-                only_mask = (True if mask == "Mask only" else False)
-            ) 
-            
-            
-            
+                session=new_session(model),
+                alpha_influence=alpha_influence,  # Control de influencia del canal alfa
+                segmentation_strength=segmentation_strength  # Control de fuerza de segmentación
+            )
+
             o.write(output)
     return os.path.join("output.png")
-        
+
 title = "RemBG"
-description = "Gradio demo for RemBG. To use it, simply upload your image and wait. Read more at the link below."
+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>"
 
-
 gr.Interface(
     inference, 
     [
@@ -52,16 +50,18 @@ gr.Interface(
             "isnet-general-use",
             "isnet-anime",
             "sam",
-            ], 
-            type="value",
-            default="isnet-general-use",
-            label="Models"
+        ], 
+        type="value",
+        default="isnet-general-use",
+        label="Models"
         ),
+        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"),
     title=title,
     description=description,
     article=article,
-    examples=[["lion.png", "Default", "u2net"], ["girl.jpg", "Default", "u2net"], ["anime-girl.jpg", "Default", "isnet-anime"]],
+    examples=[["lion.png", "Default", "u2net", 0.5, 0.5], ["girl.jpg", "Default", "u2net", 0.5, 0.5], ["anime-girl.jpg", "Default", "isnet-anime", 0.5, 0.5]],
     enable_queue=True
-    ).launch()
+).launch()