added prediction code

app.py

@@ -26,9 +26,8 @@ def preprocess_image(drawing, size=(28, 28)):
     img_array = np.expand_dims(img_array, axis=-1)
     return img_array
 
-def preprocess_and_predict(img):
+def preprocess_and_predict(image):
     model = load_model("mnist_cnn_model.h5")
-    image = np.array(img) / 255.0
     # Expand dimensions to match the input shape expected by the model
     image = np.expand_dims(image, axis=0)
     # Reshape to match the input shape expected by the model
@@ -45,28 +44,25 @@ def main():
     # Create a drawing canvas
     drawing = st_canvas(
         fill_color="rgb(0, 0, 0)",  # Background color of the canvas
-        stroke_width=10,  # Stroke width
+        stroke_width=4,  # Stroke width
         stroke_color="rgb(255, 255, 255)",  # Stroke color
         background_color="#000000",  # Background color of the canvas component
-        height=150,  # Height of the canvas
-        width=150,  # Width of the canvas
+        height=168,  # Height of the canvas
+        width=168,  # Width of the canvas
         drawing_mode="freedraw",  # Drawing mode: "freedraw" or "transform"
         key="canvas",
     )
+    predict = st.button('Predict digit')
 
     # Check if the user has drawn anything
-    if drawing is not None:
-        st.image(drawing.image_data)
+    if predict is True:
 
         # Preprocess the drawn image
         processed_image = preprocess_image(drawing.image_data)
-        st.write("Processed Image Shape:", processed_image.shape)
-
-        # Save the processed image
-        np.save("processed_image.npy", processed_image)
-        print(processed_image.shape)
         digit_class = preprocess_and_predict(processed_image)
+        st.title("Predicted Digit:")
         st.success(digit_class)
+        predict = False
 
 if __name__ == "__main__":
     main()