Update app.py

app.py

@@ -4,7 +4,6 @@ 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)
@@ -19,11 +18,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,
-                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,
+                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
                 session=new_session(model)
             )
 
@@ -31,14 +30,13 @@ def inference(file, mask, model, alpha_influence, segmentation_strength, smoothi
 
     return Image.open(BytesIO(output))
 
-def save_image(output_image):
-    output_image.save("output.png", "PNG")
-
 title = "Background Using RemBG"
-description = "Gradio demo for RemBG. Erase the background of any image by adjusting the sliders and choosing a model from the U2net library. / <a href='https://huggingface.co/spaces/KenjieDec/RemBG' target='_blank'>Model on Hugging Face</a> / <a href='https://github.com/danielgatis/rembg' target='_blank'>Github Repo</a>"
+description = "<a href='https://www.buymeacoffee.com/diego2554' target='_blank'>Help me improve my computer equipment, I need RTX 4070 :)</a>Gradio demo for RemBG. erase the background of any image, To use it, simply upload your image and adjust the sliders and choose a eraser plugin from the U2net library. / <a href='https://huggingface.co/spaces/KenjieDec/RemBG' target='_blank'>Original article made by KenjieDec</a> / <a href='https://github.com/danielgatis/rembg' target='_blank'>Github Repo</a></p>"
+article = "<p style='text-align: center;'><a href='https://github.com/danielgatis/rembg' target='_blank'>Github Repo</a></p>"
+article = "<p style='text-align: center;'><a href='https://huggingface.co/spaces/KenjieDec/RemBG' target='_blank'>Model on Hugging Face</a></p>"
 
 gr.Interface(
-    [inference, save_image],  # Lista de funciones
+    inference, 
     [
         gr.inputs.Image(type="filepath", label="Input"),
         gr.inputs.Radio(
@@ -68,11 +66,11 @@ gr.Interface(
         gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.5, label="Segmentation Strength"),
         gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.5, label="Smoothing"),
     ], 
-    [
-        gr.outputs.Image(type="PIL", label="Output"),
-    ],
+    gr.outputs.Image(type="PIL", label="Output"),
+    #description = "<a href='https://huggingface.co/spaces/KenjieDec/RemBG' target='_blank'>Original article made by KenjieDec</a><a href='https://github.com/danielgatis/rembg' target='_blank'>Github Repo</a></p>"
     title=title,
     description=description,
+    article=article,
     examples=[["lion.png", "Default", "u2net", 0.5, 0.5, 0.5], ["girl.jpg", "Default", "u2net", 0.5, 0.5, 0.5], ["anime-girl.jpg", "Default", "isnet-anime", 0.5, 0.5, 0.5]],
     enable_queue=True
 ).launch()