Update app.py

app.py

@@ -85,8 +85,8 @@ def preprocess_mri(nifti_path):
 
 # 🔹 Run Brain Age Prediction (Decorated for GPU Execution)
 @spaces.GPU(duration=90)
-def predict_brain_age(age): #, actual_age):
-    return f"Brain Age estimate: 42"
+def predict_brain_age(age, sex): #, actual_age):
+    return f"Brain Age estimate: 42", sex
 
 """
 def predict_brain_age(nifti_file, actual_age): #sex
@@ -144,7 +144,7 @@ with gr.Blocks() as demo:
         with gr.Column(scale=1):
             #mri_input = gr.File(label="Upload a T1w MRI (NIfTI .nii.gz)")
             age_input = gr.Number(label="Enter Age", value=50)
-            #sex_input = gr.Radio(["Male", "Female"], label="Select Sex")
+            sex_input = gr.Radio(["Male", "Female"], label="Select Sex")
             submit_button = gr.Button("Predict")
 
         with gr.Column(scale=2):
@@ -155,7 +155,7 @@ with gr.Blocks() as demo:
     
     submit_button.click(
         fn=predict_brain_age,
-        inputs=[age_input], # mri_input, sex_input
+        inputs=[age_input, sex_input], # mri_input, sex_input
         outputs=[brain_age_output, bad_output]
     )