Update app.py

app.py

@@ -17,6 +17,8 @@ def predict(image_array):
             
         image = Image.fromarray(image_array.astype("uint8"), mode="L")
         image = image.resize((28, 28))
+        image = image.point(lambda x: 0 if x < 128 else 255, 'L')
+        image = image.resize((28, 28))
         image_array = np.array(image).astype("float32") / 255.0
         image_array = image_array.reshape(1, 28, 28, 1)
         logits = model.predict(image_array)