add: fix prediction shape

app.py

@@ -61,6 +61,7 @@ def get_results(image):
 	plt.axis("off")
 
 	prediction = tf.nn.softmax(logits, axis=-1)
+	prediction = prediction.numpy()[0]
 
 	return plt, {labels[i]: float(prediction[i]) for i in range(10)}