Update app.py

app.py

@@ -85,9 +85,12 @@ def preprocess_mri(nifti_path):
 
 # 🔹 Run Brain Age Prediction (Decorated for GPU Execution)
 @spaces.GPU(duration=90)
-def predict_brain_age(path, age, sex): #, actual_age):
+def predict_brain_age_core(path, age, sex): #, actual_age):
     return f"Brain Age estimate: 42", age
 
+def predict_brain_age(nifti_file, age, sex):
+    return predict_brain_age_core(nifti_file.name, age, sex)
+
 """
 def predict_brain_age(nifti_file, actual_age): #sex
     if not os.path.exists(nifti_file.name):
@@ -142,11 +145,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)")
-            flair_input = gr.File(label="Upload a FLAIR Image (.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)")
             submit_button = gr.Button("Predict")
 
         with gr.Column(scale=2):
@@ -157,7 +158,7 @@ with gr.Blocks() as demo:
     
     submit_button.click(
         fn=predict_brain_age,
-        inputs=[flair_input, age_input, sex_input],
+        inputs=[mri_input, age_input, sex_input],
         outputs=[brain_age_output, bad_output]
     )