Update app.py

app.py

@@ -4,6 +4,7 @@ import cv2
 from rembg import new_session, remove
 from PIL import Image
 from io import BytesIO
+import shutil
 
 def inference(file, mask, model, alpha_influence, segmentation_strength, smoothing):
     im = cv2.imread(file, cv2.IMREAD_COLOR)
@@ -18,11 +19,11 @@ def inference(file, mask, model, alpha_influence, segmentation_strength, smoothi
             output = remove(
                 input, 
                 only_mask=(True if mask == "Mask only" else False),
-                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
-                alpha_matting_smoothing=smoothing,  # Control de suavizado de bordes de la segmentación
+                alpha_matting=True,
+                alpha_matting_foreground_threshold=alpha_influence,
+                alpha_matting_background_threshold=1 - alpha_influence,
+                alpha_matting_erode_size=int(segmentation_strength * 10),
+                alpha_matting_smoothing=smoothing,
                 session=new_session(model)
             )
 
@@ -69,7 +70,6 @@ gr.Interface(
     ], 
     [
         gr.outputs.Image(type="PIL", label="Output"),
-        gr.outputs.Button(label="Download Output", type="data")
     ],
     title=title,
     description=description,