Update app.py

app.py

@@ -46,20 +46,21 @@ canvas_result = st_canvas(
     stroke_width=3,
     stroke_color="rgba(0, 0, 0, 1)",
     background_color="rgba(255, 255, 255, 1)",
-    height=300,
-    width=500,
+    height=128,
+    width=128,
     update_streamlit=True,
     key="canvas"
 )
 
-# Add a clear button
-if st.button("Clear Canvas"):
-    canvas_result.delete("all")
+
 
 # Add a submit button
 if st.button("Submit"):
     if canvas_result.json_data:
-        st.text(f"Image Details: {canvas_result}")
+        image_from_canvas = canvas_result["image_data"]
+        preprocessed_image = preprocess_image(image_from_canvas)
+        predicted_label = predict(preprocessed_image)
+        st.text(f"Predicted label: {predicted_label}")
     else:
         st.text("Please draw an image on the canvas before submitting.")