Update app.py

app.py

@@ -60,7 +60,7 @@ with gr.Blocks() as demo:
     
     # train the model - 5 runs
     # evaluate the model on the test set
-    model.fit(x_train, y_train, epochs=5, validation_split=0.9)
+    model.fit(x_train, y_train, epochs=2, validation_split=0.9)
     test_loss, test_acc = model.evaluate(x_test,  y_test, verbose=2)
     post_train_results = f"Test accuracy: {test_acc} Test Loss: {test_loss}"
     print(post_train_results)