Update app.py

app.py

@@ -3,6 +3,7 @@ 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",
@@ -19,6 +20,8 @@ def predict(image):
     bg_color = (0, 0, 0)  # Color de fondo predeterminado (negro)
     try:
         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)  # Convertir el array de numpy a una imagen PIL
         return result
     except ValueError as err:
         logging.error(err)