Update app.py

app.py

@@ -10,6 +10,13 @@ from monai.transforms import Compose, LoadImaged, Spacingd, CropForegroundd, Spa
 from monai.data import Dataset
 from nnunet_mednext import create_mednext_encoder_v1
 
+# Debugging GPU environment
+if torch.cuda.is_available():
+    print(f"GPU is available: {torch.cuda.get_device_name(0)}")
+else:
+    print("No GPU available. Falling back to CPU.")
+    os.system("nvidia-smi")
+
 # Device selection
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
@@ -101,7 +108,7 @@ def predict_brain_age(mri_path, actual_age, sex):
            f"Brain Age Difference (BAD): {brain_age_difference:.2f} years"
 
 # Gradio UI
-iface = gr.Interface(
+demo = gr.Interface(
     fn=predict_brain_age,
     inputs=[
         gr.File(label="Upload MRI (NIfTI .nii.gz)"),
@@ -119,4 +126,4 @@ iface = gr.Interface(
 
 # Launch the Gradio app
 if __name__ == "__main__":
-    iface.launch(share=True)
+    demo.launch(share=True)