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, sex): #, actual_age):
-    return f"Brain Age estimate: 42", sex
+def predict_brain_age(path, age, sex): #, actual_age):
+    return f"Brain Age estimate: 42", age
 
 """
 def predict_brain_age(nifti_file, actual_age): #sex
@@ -143,6 +143,8 @@ 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)")
+            mri_input = gr.File(label="Upload a T1w MRI", type="filepath", file_types=[".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")
@@ -155,7 +157,7 @@ with gr.Blocks() as demo:
     
     submit_button.click(
         fn=predict_brain_age,
-        inputs=[age_input, sex_input], # mri_input, sex_input
+        inputs=[path, age_input, sex_input], # mri_input, sex_input
         outputs=[brain_age_output, bad_output]
     )