Update app.py

app.py

@@ -54,6 +54,7 @@ try:
     st.image(image, width=256, caption='Uploaded Image')
     if st.button('Predict'):
 
+      img = labels[np.argmax(prediction[0], axis=-1)])
 
       print("---------------img-array---------------------")
       print(img[np.newaxis, ...])