Update

app.py

@@ -8,7 +8,7 @@ model = tf.keras.models.load_model("best_model_mobilenetv2_finetune.h5")
 def process(img):
     img = img.resize((224,224))
     img = np.array(img) / 255.0
-    return np.expand_dims(img, axis=1)
+    return np.expand_dims(img, axis=0)
 
 def predict(img):
     img = process(img)