Update app.py

app.py

@@ -137,6 +137,7 @@ def transform_image(image):
 
 # Definimos una función que se encarga de llevar a cabo las predicciones
 def predict(img):
+    img=Image.fromarray(img)
     image = transforms.Resize((480,640))(img)
     tensor = transform_image(image=image)
     with torch.no_grad():