fix accuracy display

app.py

@@ -16,10 +16,12 @@ def predict_image(image):
 
   predictions = model.signatures['serving_default'](tf.convert_to_tensor(image_array, dtype=tf.float32))['output_0']
 
+  probabilities = tf.nn.softmax(predictions, axis=1).numpy()
+    
   # Highest prediction
-  top_index = np.argmax(predictions.numpy(), axis=1)[0]
+  top_index = np.argmax(probabilities.numpy(), axis=1)[0]
   top_label = labels[top_index]
-  top_probability = predictions.numpy()[0][top_index] * 100
+  top_probability = probabilities.numpy()[0][top_index] * 100
 
   return {top_label:top_probability}