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(nifti_file): #, actual_age):
     return f"Brain Age estimate: 42"
 
 """
@@ -142,9 +142,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 (NIfTI .nii.gz)", file_types=[".nii.gz"], type="filepath")
-            age_input = gr.Number(label="Enter Age", value=50)
+            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")
 
@@ -156,7 +155,7 @@ with gr.Blocks() as demo:
     
     submit_button.click(
         fn=predict_brain_age,
-        inputs=[mri_input, age_input], #sex_input
+        inputs=[mri_input], # age_input, sex_input
         outputs=[brain_age_output, bad_output]
     )