Update app.py

app.py

@@ -19,9 +19,9 @@ def predict_image(image):
   probabilities = tf.nn.softmax(predictions, axis=1).numpy()
     
   # Highest prediction
-  top_index = np.argmax(probabilities.numpy(), axis=1)[0]
+  top_index = np.argmax(probabilities, axis=1)[0]
   top_label = labels[top_index]
-  top_probability = probabilities.numpy()[0][top_index] * 100
+  top_probability = probabilities[0][top_index] * 100
 
   return {top_label:top_probability}