Update app.py

app.py

@@ -148,10 +148,10 @@ canvas_result = st_canvas(
 if canvas_result.image_data is not None:
     image = canvas_result.image_data
     image1 = image.copy()
-    image1 = image1.astype('float32')
-    image1 = cv2.cvtColor(image1, cv2.COLOR_BGR2GRAY)
+    image1 = cv2.cvtColor(image1, cv2.COLOR_BGR2GRAY).astype('float32')  # Convert to float32
     image1 = cv2.resize(image1, (28, 28))
-    image1 = (image1 * 255).astype('uint8')  # Convert to uint8 format (bytes)
+    image1 /= 255.0  # Normalize the values to [0, 1]
+     # Convert to uint8 format (bytes)
     st.image(image1)