Update app.py

app.py

@@ -4,15 +4,11 @@ from PIL import Image
 import numpy as np
 
 # Load your custom regression model
-model_path = "kia_mnist_keras_model.weights.h5"  
-model = tf.keras.Sequential([
-    tf.keras.layers.Flatten(input_shape=[28, 28]),
-    tf.keras.layers.Rescaling(1./255.),
-    tf.keras.layers.Dense(300, activation="relu"),
-    tf.keras.layers.Dense(100, activation="relu"),
-    tf.keras.layers.Dense(10, activation="softmax")
-])
-model.load_weights(model_path)
+model_path = "kia_mnist_keras_model.weights.h5"
+model_path = "kia_mnist_keras_model.keras"
+
+#model.load_weights(model_path)
+model = tf.keras.models.load_model(model_path)
 
 labels = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9']