Update app.py

app.py

@@ -43,6 +43,14 @@ with gr.Blocks() as demo:
         test_loss, test_acc = model.evaluate(x_test,  y_test, verbose=2)
 
         return('\nTest accuracy:', test_acc)
+
+        def predict_image(img):
+        # Define any necessary preprocessing steps for the image input here
+        # Make a prediction using the model
+        prediction = model.predict(preprocessed_img)
+
+        # Postprocess the prediction and return it
+        return prediction
         
 
     # Creates the Gradio interface objects
@@ -50,8 +58,11 @@ with gr.Blocks() as demo:
         with gr.Column(scale=1):
             submit_btn = gr.Button(value="Train/Eval")
         with gr.Column(scale=2):
-            model_data = gr.Text(label="Model Results", interactive=False)
-    submit_btn.click(modelTraining, [], model_data)
+            model_performance = gr.Text(label="Model Performance", interactive=False)
+            model_prediction = gr.Text(label="Model Prediction", interactive=False)
+            image_data = gr.Image(label="Upload Image")
+    submit_btn.click(modelTraining, [], model_performance)
+    image_data.change(predict_image, image_data, model_prediction)
     
     
 # creates a local web server