Update app.py

app.py

@@ -1,8 +1,7 @@
 import gradio as gr
 import os
 import cv2
-from rembg import new_session, remove
-from RemBG_super/rembg/bg import remove as bg_remove  # Importa la función 'remove' desde bg.py
+from bg import remove
 
 def inference(file, mask, model, alpha_influence, segmentation_strength):
     im = cv2.imread(file, cv2.IMREAD_COLOR)
@@ -14,22 +13,19 @@ def inference(file, mask, model, alpha_influence, segmentation_strength):
     with open(input_path, 'rb') as i:
         with open(output_path, 'wb') as o:
             input = i.read()
-            if mask == "Mask only":
-                output = bg_remove(input, only_mask=True)
-            else:
-                session = new_session(model)
-                output = remove(
-                    input,
-                    session=session,
-                    alpha_influence=alpha_influence,
-                    bg_color=(0, 0, 0, segmentation_strength)
-                )
+            output = remove(
+                input, 
+                session=new_session(model), 
+                only_mask=(True if mask == "Mask only" else False),
+                alpha=alpha_influence,
+                bg_color=(0, 0, 0, segmentation_strength)
+            )
 
             o.write(output)
     return os.path.join("output.png")
 
-title = "RemBG with bg.py"
-description = "Gradio demo for RemBG using bg.py. To use it, simply upload your image and adjust the alpha influence and segmentation strength."
+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>"
 
 gr.Interface(