Update app.py

app.py

@@ -102,7 +102,7 @@ class TransformerBlock(tf.keras.layers.Layer):
 
 # Transformer Model
 def TransformerModel(trainX, trainy, testX, testy, embed_dim=128, num_heads=8, ff_dim=256, 
-                      epochs=100, batch_size=64, learning_rate=0.0001, dropout_rate=0.3):
+                      epochs=1, batch_size=64, learning_rate=0.0001, dropout_rate=0.3):
 
     # Feature Scaling
     scaler = MinMaxScaler()
@@ -323,7 +323,7 @@ def run_cross_validation(training_file, training_additive_file, testing_file, te
                          training_dominance_file, testing_dominance_file, feature_selection, learning_rate, min_child_weight):
 
     # Default parameters
-    epochs = 1000
+    epochs = 1
     batch_size = 64
     outer_n_splits = 2