Update app.py

app.py

@@ -3,12 +3,12 @@ import tensorflow as tf
 import numpy as np
 
 # Load the model
-model = tf.keras.saving.load_model('mnist_trained_model_2(acc-97%).h5')
+model = tf.keras.saving.load_model('mnist_model.h5')
 
 # Classification prediction function
 labels = [0,1,2,3,4,5,6,7,8,9]
 def classify_image(image):
-  prediction = model.predict(image.reshape(-1,784)/255).flatten()
+  prediction = model.predict(image.reshape(28,28))
   confidences = {labels[i]: float(prediction[i]) for i in range(10)}
   return confidences