Update app.py

app.py

@@ -37,9 +37,11 @@ def predict_digit(input_data):
     params = np.load("trained_params.npz", allow_pickle=True)
     W1, b1, W2, b2 = params["W1"], params["b1"], params["W2"], params["b2"]
 
-    
-    # Reshape the input data to match the neural network architecture
-    X = input_data.reshape((784, 1)) / 255.
+    # Convert the uploaded image to grayscale and resize it to (28, 28)
+    img = Image.fromarray(input_data).convert("L").resize((28, 28))
+
+    # Convert the image to a NumPy array and normalize it
+    X = np.array(img).reshape((784, 1)) / 255.
 
     # Get the prediction
     prediction = make_predictions(X, W1, b1, W2, b2)