adding a dict of prediction probabilities along with class names

app.py

@@ -48,9 +48,11 @@ def infer(input_image):
     print(image_tensor.shape)
     predictions = model.predict(np.expand_dims((image_tensor), axis=0))
     predictions = np.squeeze(predictions)
-    predictions = np.argmax(predictions) # , axis=2
-    predicted_label = class_names[predictions.item()]
-    return str(predicted_label)
+    # predictions = np.argmax(predictions) # , axis=2
+    # predicted_label = class_names[predictions.item()]
+    # return str(predicted_label)
+
+    return dict(zip(class_names, predictions))
     
     
 # get the inputs