Update app.py

app.py

@@ -154,7 +154,7 @@ with gr.Blocks() as demo:
             bad_output = gr.Textbox(label="Brain Age Difference", interactive=False)
     
     submit_button.click(
-        fn=predict_brain_age_wrapper,
+        fn=predict_brain_age,
         inputs=[mri_input, age_input], #sex_input
         outputs=[brain_age_output, bad_output]
     )