Update app.py

app.py

@@ -147,11 +147,10 @@ with gr.Blocks() as demo:
             submit_button = gr.Button("Predict")
 
         with gr.Column(scale=2):
-            brain_age_output = gr.Textbox(label="Predicted Brain Age", interactive=False)
-            bad_output = gr.HTML(label="Brain Age Difference")  # Use gr.HTML for colored text
-            #bad_output = gr.HTML(value="", show_label=True, label="Brain Age Difference")
-            #bad_output = gr.Textbox(label="Brain Age Difference", interactive=False)
-
+            #brain_age_output = gr.Textbox(label="Predicted Brain Age", interactive=False)
+            #bad_output = gr.HTML(label="Brain Age Difference")  # Use gr.HTML for colored text
+            brain_age_output = gr.Textbox(label="Output")
+            submit_button.click(fn=predict_brain_age, inputs=[mri_input, age_input], outputs=[brain_age_output])
 
     submit_button.click(
         fn=predict_brain_age,