Update app.py

app.py

@@ -3,15 +3,8 @@ import tensorflow as tf
 import numpy as np
 
 
-model_path = "iris_mlp.weights.h5"
-model = tf.keras.Sequential([
-    tf.keras.layers.InputLayer(input_shape=[4]),
-    tf.keras.layers.BatchNormalization(),
-    tf.keras.layers.Dense(32, activation="relu"),
-    tf.keras.layers.Dense(16, activation="relu"),
-    tf.keras.layers.Dense(3, activation="softmax")
-])
-model.load_weights(model_path)
+model_path = "iris_mlp.keras"
+model = tf.keras.models.load(model_path)
 
 labels = ['Setosa', 'Versicolour', 'Virginica']