Update app.py

app.py

@@ -84,7 +84,7 @@ def preprocess_mri(nifti_path):
     return next(iter(dataloader))["image"].to(device)
 
 # 🔹 Run Brain Age Prediction (Decorated for GPU Execution)
-#@spaces.GPU(duration=90)
+@spaces.GPU(duration=90)
 def predict_brain_age(nifti_file, actual_age):
     return f"Brain Age estimate: 42"
 """
@@ -155,7 +155,7 @@ with gr.Blocks() as demo:
     submit_button.click(
         fn=predict_brain_age,
         inputs=[mri_input, age_input], #sex_input
-        outputs=[brain_age_output, bad_output]
+        #outputs=[brain_age_output, bad_output]
     )
     
     gr.Markdown("""