Update modelutil.py

modelutil.py

@@ -2,10 +2,8 @@ import tensorflow as tf
 
 def create_model():
 
-    LAYERS = [
-              tf.keras.layers.Flatten(input_shape=[28,28], name="inputlayer"),
+    LAYERS = [tf.keras.layers.Flatten(input_shape=[28,28], name="inputlayer"),
               tf.keras.layers.Dense(300, activation='relu', name="hiddenlayer1"),
-              tf.keras.layers.Dense(256, activation='relu', name="hiddenlayer1"),
               tf.keras.layers.Dense(128, activation='relu', name="hiddenlayer2"),
               tf.keras.layers.Dense(10, activation='softmax', name="outputlayer")]