Update app.py

app.py

@@ -87,6 +87,10 @@ def preprocess_mri(nifti_path):
 @spaces.GPU(duration=90)
 def predict_brain_age(nifti_file, actual_age):
     return f"Brain Age estimate: 42"
+
+def predict_brain_age_wrapper(nifti_file: gr.File, actual_age: float) -> tuple[str, str]:
+    return predict_brain_age(nifti_file, actual_age)
+    
 """
 def predict_brain_age(nifti_file, actual_age): #sex
     if not os.path.exists(nifti_file.name):
@@ -153,9 +157,9 @@ with gr.Blocks() as demo:
             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]
+        fn=predict_brain_age_wrapper,
+        inputs=[mri_input, age_input], #sex_input
+        outputs=[brain_age_output, bad_output]
     )
     
     gr.Markdown("""