Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import os
 import cv2
+import numpy as np
 
 def inference(file, mask, model, alpha_influence, segmentation_strength):
     im = cv2.imread(file, cv2.IMREAD_COLOR)
@@ -20,15 +21,15 @@ def inference(file, mask, model, alpha_influence, segmentation_strength):
                     input, 
                     session=new_session(model), 
                     only_mask=(True if mask == "Mask only" else False),
-                    alpha=alpha_influence,
-                    bg_color=(0, 0, 0, segmentation_strength)
+                    alpha=alpha_influence,  # Control de influencia del canal alfa
+                    bg_color=(0, 0, 0, segmentation_strength)  # Control de fuerza de segmentación
                 )
                 o.write(output)
                 m.write(output)
 
     return os.path.join("output.png"), os.path.join("mask.png")
 
-title = "RemBG_Super"
+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>"
 
@@ -71,8 +72,8 @@ iface = gr.Interface(
         gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.5, label="Segmentation Strength"),
     ], 
     [
-        gr.outputs.Image(type="plot", label="Processed Image"),
-        gr.outputs.Image(type="plot", label="Processed Mask"),
+        gr.outputs.Image(type="plot", label="Processed Image", output=show_processed_image),
+        gr.outputs.Image(type="plot", label="Processed Mask", output=show_processed_mask),
     ],
     title=title,
     description=description,