Update app.py

app.py

@@ -153,7 +153,7 @@ with gr.Blocks() as demo:
             brain_age_output = gr.Textbox(label="Predicted Brain Age", interactive=False)
             bad_output = gr.Textbox(label="Brain Age Difference", interactive=False)
     
-    submit_button.click(fn=predict_brain_age, inputs=[mri_input, age_input, sex_input], outputs=[brain_age_output, bad_output])
+    submit_button.click(fn=predict_brain_age, inputs=[age_input, age_input, sex_input], outputs=[brain_age_output, bad_output])
     
     gr.Markdown("""
     **Disclaimer:** This is a research tool and is not intended for clinical use.