Update app.py

app.py

@@ -21,7 +21,7 @@ mnist_examples = [[cv2.resize(x_test[i], DIMS)] for i in range(10)]
 # Function to preprocess the image
 def preprocess_image(image):
     if isinstance(image, Image.Image):
-        image = np.array(image)
+        image = np.array(image.convert('L'))  # Convert to grayscale
     image = cv2.resize(image, (28, 28))  # Resize to 28x28 for model input
     image = image / 255.0
     image = image.reshape(1, 28, 28, 1)