Update app.py

app.py

@@ -1,4 +1,5 @@
 import logging
+import numpy as np
 import gradio as gr
 from rembg import new_session
 from cutter import remove, make_label
@@ -21,7 +22,7 @@ def predict(image):
     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
+            result = Image.fromarray(result.astype('uint8'))  # Convertir el array de numpy a una imagen PIL
         return result
     except ValueError as err:
         logging.error(err)