Update app.py

app.py

@@ -4,7 +4,6 @@ import gradio as gr
 from rembg import new_session
 from cutter import remove, make_label
 from utils import *
-from PIL import Image
 
 remove_bg_models = {
     "U2NET": "u2net",
@@ -18,15 +17,15 @@ def predict(image):
     session = new_session(remove_bg_models[default_model])
     smoot = False
     matting = (0, 0, 0)  # Valores predeterminados para matting
-    bg_color = (0, 0, 0)  # Color de fondo predeterminado (negro)
+    bg_color = False  # Color de fondo predeterminado (no cambiar color)
     try:
-        result = remove(session, image, smoot, matting, bg_color)
+        result, _ = remove(session, image, smoot, matting, bg_color)
         if isinstance(result, np.ndarray):  # Verificar si la salida es un array de numpy
             result = Image.fromarray(result.astype('uint8'))  # Convertir el array de numpy a una imagen PIL
         return result
     except ValueError as err:
         logging.error(err)
-        return None
+        return make_label(str(err)), None
 
 with gr.Blocks(title="Remove background") as app:
     gr.HTML("<center><h1>Background Remover</h1></center>")
@@ -35,7 +34,7 @@ with gr.Blocks(title="Remove background") as app:
             input_img = gr.Image(type="pil", label="Input image")
         with gr.Column():
             output_img = gr.Image(type="pil", label="Result image")
-
+    
     with gr.Row(equal_height=True):
         run_btn = gr.Button(value="Remove background", variant="primary")
         clear_btn = gr.Button(value="Clear", variant="secondary")