Update app.py

app.py

@@ -146,15 +146,15 @@ canvas_result = st_canvas(
 
 # Do something interesting with the image data and paths
 if canvas_result.image_data is not None:
-    #st.image(canvas_result.image_data)
     image = canvas_result.image_data
     image1 = image.copy()
     image1 = image1.astype('float32')
-    image1 = cv2.cvtColor(image1,cv2.COLOR_BGR2GRAY)
-    image1 = cv2.resize(image1,(28,28))
-    image1 = image1 / 255.0
+    image1 = cv2.cvtColor(image1, cv2.COLOR_BGR2GRAY)
+    image1 = cv2.resize(image1, (28, 28))
+    image1 = (image1 * 255).astype('uint8')  # Convert to uint8 format (bytes)
     st.image(image1)
 
+
     image1.resize(1,1,28,28)
     st.title(np.argmax(cnn.predict(image1)))
 if canvas_result.json_data is not None: