Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 import os
 import cv2
-import bg  # Importa el módulo "bg.py"
 
 def inference(file, mask, model, alpha_influence, segmentation_strength):
     im = cv2.imread(file, cv2.IMREAD_COLOR)
@@ -17,16 +16,16 @@ def inference(file, mask, model, alpha_influence, segmentation_strength):
             input = i.read()
             output = remove(
                 input, 
-                session=bg.new_session(model),  # Utiliza la función de "bg.py"
+                session=new_session(model), 
                 only_mask=(True if mask == "Mask only" else False),
-                alpha_influence=alpha_influence,  # Control de influencia del canal alfa
-                segmentation_strength=segmentation_strength  # Control de fuerza de segmentación
+                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)
     return os.path.join("output.png")
 
-title = "RemBG"
+title = "RemBG_ Super"
 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>"