Update app.py

app.py

@@ -13,6 +13,7 @@ examples = examples = [
 def inference(original_image):
     image = tf.keras.utils.img_to_array(original_image)
     image = image.astype("float32") / 255.0
+    image = image[:,:,0]
     image = np.reshape(image, (28, 28, 1))
     output = model.predict(image)
     return output