Update app.py

app.py

@@ -251,6 +251,10 @@ def predict(image):
     # Normalize the image
     mean = [0.485, 0.456, 0.406]
     std = [0.229, 0.224, 0.225]
+    
+    mean = np.expand_dims(mean, axis=(1,2))
+    std = np.expand_dims(std, axis=(1,2))
+    
     image = (image / 255.0 - mean)/std
 
     print("normalized image:", image.shape)