Updated app.py

app.py

@@ -71,13 +71,12 @@ interface = gr.Interface(
     description="Upload a skull-stripped FLAIR image (.nii.gz) to generate a binary segmentation of MS lesions."
 )
 
-# Force GPU initialization
+# Debugging GPU environment
 if torch.cuda.is_available():
-    print("CUDA is available. Initializing GPU...")
-    device = torch.device("cuda:0")
-    torch.tensor([1.0]).to(device)  # Trigger GPU initialization
+    print(f"GPU is available: {torch.cuda.get_device_name(0)}")
 else:
     print("No GPU available. Falling back to CPU.")
+    os.system("nvidia-smi")  # Check if NVIDIA tools are available
 
 # Download model files before launching the app
 download_model()