Update app.py

app.py

@@ -28,6 +28,12 @@ import re # <--- ADD THIS LINE FOR THE NAMEERROR
 # Set CUDA_HOME explicitly for custom CUDA kernel compilation
 os.environ["CUDA_HOME"] = "/usr/local/cuda"
 
+
+print(f"Is CUDA available: {torch.cuda.is_available()}")
+# True
+print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+# Tesla T4
+
 # Flag to check if GPU is present
 HAS_GPU = False # Initialize to False, let pynvml determine
 GPU_COUNT = 0