Update app.py

app.py

@@ -9,6 +9,12 @@ import torch
 # Check if CUDA (GPU) is available
 print("CUDA available:", torch.cuda.is_available())
 
+if torch.cuda.is_available():
+    print("GPU Name:", torch.cuda.get_device_name(0))
+else:
+    print("No GPU detected. Running on CPU.")
+
+    
 # Set a seed for reproducibility
 set_seed(42)