Update app.py

app.py

@@ -22,7 +22,7 @@ def import_and_predict(image_data, model):
     img = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
     #img_resize = (cv2.resize(img, dsize=(75, 75),    interpolation=cv2.INTER_CUBIC))/255.
     
-    img_reshape = img[np.newaxis,...]
+    img_reshape = np.reshape(img,(1,224,224,3))
     
     st.write(img_reshape.shape)