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=2, validation_split=0.9)
+    model.fit(x_train, y_train, epochs=5, validation_split=0.3)
     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)
@@ -89,7 +89,7 @@ with gr.Blocks() as demo:
     # Creates the Gradio interface objects
     with gr.Row():
         with gr.Column(scale=2):
-            image_data = gr.Image(label="Upload Image", type="numpy", image_mode="L", shape=[28,28], invert_colors=True)
+            image_data = gr.Image(label="Upload Image")
         with gr.Column(scale=1):
             model_prediction = gr.Text(label="Model Prediction", interactive=False)
         image_data.change(classifyImage, image_data, model_prediction)