Update app.py

app.py

@@ -85,7 +85,7 @@ def preprocess_mri(nifti_path):
 
 # 🔹 Run Brain Age Prediction (Decorated for GPU Execution)
 @spaces.GPU(duration=90)
-def predict_brain_age(nifti_file): #, actual_age):
+def predict_brain_age(age): #, actual_age):
     return f"Brain Age estimate: 42"
 
 """
@@ -142,9 +142,9 @@ with gr.Blocks() as demo:
           
     with gr.Row():
         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")
+            #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")
             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=[mri_input], # age_input, sex_input
+        inputs=[age_input], # mri_input, sex_input
         outputs=[brain_age_output, bad_output]
     )