Update app.py

app.py

@@ -111,7 +111,7 @@ if canvas_result.image_data is not None:
 
     # Correctly reshape the image
     image1 = image1.reshape(1, 1, 28, 28).astype('float32')
-    prediction = np.argmax(cnn.predict(image1))
+    prediction = cnn.predict(image1)
     st.title(f"Handwritten Digit Prediction: {prediction}")