Update app.py

app.py

@@ -75,7 +75,7 @@ def predict_image(input):
     predictions = model1.predict(image_array)
     
     # Get the predicted class label
-    predicted_class_index = np.argmax(predictions, -1)
+    predicted_class_index = tf.argmax(predictions, axis=1).numpy()
     predicted_class_label = class_labels[predicted_class_index]
     
     # Get the confidence score of the predicted class