predictions

app.py

@@ -61,10 +61,10 @@ def on_click():
         st.image(image, use_column_width=True)
         image = image.convert('RGB')
         predictions = model_prediction(np.array(image))
-        st.write(prediction)
+        st.write(predictions)
         print(
             "This image most likely belongs to {}."
-            .format(prediction)
+            .format(predictions)
 )
 
 st.button('Predict', on_click=on_click)