Adding debug messages

app.py

@@ -4,6 +4,12 @@ import subprocess
 import time
 import signal
 import atexit
+import torch
+
+print("CUDA Available:", torch.cuda.is_available())
+print("GPU Count:", torch.cuda.device_count())
+print("Device Name:", torch.cuda.get_device_name(0) if torch.cuda.is_available() else "No GPU")
+
 
 SYSTEM_PROMPT = "You are a compliance assistant. Use the provided risk data to answer user questions. If a single risk object is given, provide a direct answer. If a list of risks is provided, summarize, compare, or analyze the collection as needed. Always base your response on the data provided."
 hf_token = os.environ["HF_TOKEN"]