Update app.py

app.py

@@ -40,14 +40,16 @@ with gr.Blocks() as demo:
         
         
         model.fit(x_train, y_train, epochs=5)
-        test_loss, test_acc = model.evaluate(x_test,  y_test, verbose=2)
+        test_loss, test_acc = model.evaluate(x_test,  y_test, verbose=2)]
 
-        return('\nTest accuracy:', test_acc)
+        result = "Test accuracy: ", test_acc
+
+        return result
 
         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)
+        prediction = model.predict(img)
 
         # Postprocess the prediction and return it
         return prediction
@@ -60,7 +62,7 @@ with gr.Blocks() as demo:
         with gr.Column(scale=2):
             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")
+            image_data = gr.Image(label="Upload Image", type="numpy")
     submit_btn.click(modelTraining, [], model_performance)
     image_data.change(predict_image, image_data, model_prediction)