Update files/inference.py

files/inference.py

@@ -69,6 +69,10 @@ def predict():
 
             # Aplicar la transformación a la imagen
             image = Image.open(file)
+            if image.mode in ("RGBA", "P"):
+                # Convert the image to RGB mode
+                image = image.convert("RGB")
+            
             image_tensor = transform(image).unsqueeze(0)
 
             # Obtener la salida del modelo