Update app.py

app.py

@@ -25,7 +25,9 @@ def transform_image(image):
 
 # Definimos una función que se encarga de llevar a cabo las predicciones
 def predict(img):
-    image = transforms.Resize((480,640))(img)
+    img_pil = PIL.Image.fromarray(img, 'RGB')
+    
+    image = transforms.Resize((480,640))(img_pil)
     tensor = transform_image(image=image)
 
     model.to(device)