Update app.py

app.py

@@ -10,6 +10,7 @@ with open(join(current_dir, 'image_labels.json')) as labels_file:
 
 mobile_net = tf.keras.applications.MobileNetV2()
 def image_classifier(img):
+    img = cv2.resize(img, (224,224))
     arr = np.expand_dims(img, axis=0)
     arr = tf.keras.applications.mobilenet.preprocess_input(arr)
     prediction = mobile_net.predict(arr).flatten()