Update app.py

app.py

@@ -155,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]
     )
     
@@ -179,20 +179,9 @@ else:
     os.system("nvidia-smi")
     device = torch.device("cpu")
     
-# 🔹 Download Model Weights
+#  Download Model Weights
 download_model()
 
-# 🔹 Run Gradio App
-#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."
-)
-
+#  Run Gradio App
 if __name__ == "__main__":
-    iface.launch(share=True)
+    demo.launch(share=True)