added comments and expander of the resized image

app.py

@@ -36,17 +36,21 @@ canvas_result = st_canvas(
 
 if canvas_result.image_data is not None:
 
+    # Get image data from canvas
     im = ImageOps.grayscale(Image.fromarray(canvas_result.image_data.astype(
         'uint8'), mode="RGBA")).resize((28, 28))
 
+    # Convert image to array and reshape
     data = img_to_array(im)
     data = data / 255
     data = data.reshape(1, 28, 28, 1)
     data = data.astype('float32')
 
+    # Predict digit
     st.write('### Predicted Digit')
     prediction = model.predict(data)
 
+    # Plot prediction
     result = plt.figure(figsize=(12, 3))
     plt.bar(range(10), prediction[0])
     plt.xticks(range(10))
@@ -56,6 +60,8 @@ if canvas_result.image_data is not None:
     plt.ylim(0, 1)
     st.write(result)
 
-    st.write('### Resized Image')
-    st.write("The image needs to be resized, because it can only input 28x28 images")
-    st.image(im, width=28*9)
+    # Show resized image
+    with st.expander('Show Resized Image'):
+        st.write("The image needs to be resized, because it can only input 28x28 images")
+        st.image(im, caption='Resized Image', width=28*9)
+