Update app.py

app.py

@@ -37,10 +37,12 @@ def predict_digit(img):
     W1, b1, W2, b2 = params["W1"], params["b1"], params["W2"], params["b2"]
 
     # Convert the sketchpad drawing to grayscale and resize it to (28, 28)
-    img_pil = Image.fromarray(np.uint8(img * 255)).convert("L").resize((28, 28))
+    img_pil = Image.fromarray(np.uint8(img * 255)).convert("L")
 
+    res=img_pil.resize((28, 28))
+    
     # Convert the image to a NumPy array and normalize it
-    X = np.array(img_pil).reshape((784, 1)) / 255.
+    X = np.array(res).reshape((784, 1)) / 255.
 
     # Get the prediction
     prediction = make_predictions(X, W1, b1, W2, b2)