change prediction display

app.py

@@ -15,13 +15,11 @@ def predict_image(image):
   image_array = np.expand_dims(image_array, axis=0)
 
   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(probabilities, axis=1)[0]
+  top_index = np.argmax(predictions.numpy(), axis=1)[0]
   top_label = labels[top_index]
-  top_probability = probabilities[0][top_index] * 100
+  top_probability = predictions.numpy()[0][top_index]
 
   return {top_label:top_probability}