Update app.py

app.py

@@ -88,9 +88,6 @@ def preprocess_mri(nifti_path):
 def predict_brain_age(nifti_file, actual_age):
     return f"Brain Age estimate: 42"
 
-def predict_brain_age_wrapper(nifti_file: gr.File, actual_age: float) -> tuple[str, str]:
-    return predict_brain_age(nifti_file, actual_age)
-    
 """
 def predict_brain_age(nifti_file, actual_age): #sex
     if not os.path.exists(nifti_file.name):
@@ -186,6 +183,16 @@ else:
 download_model()
 
 # 🔹 Run Gradio App
-if __name__ == "__main__":
-    demo.launch(share=True)
+#if __name__ == "__main__":
+#    demo.launch(share=True)
 
+iface = gr.Interface(
+    fn=predict_brain_age,  # decorated with @spaces.GPU
+    inputs=[gr.File(label="Upload T1w MRI"), gr.Number(label="Enter Age")],
+    outputs=[gr.Textbox(label="Predicted Brain Age"), gr.Textbox(label="Brain Age Difference")],
+    title="🧠 BrainAgeNeXt",
+    description="Upload a T1w MRI and enter the subject's age to predict brain age."
+)
+
+if __name__ == "__main__":
+    iface.launch(share=True)