handle no GPU

app.py

@@ -26,7 +26,8 @@ model = AutoModelForCausalLM.from_pretrained(
 )
 model = model.to("cuda:0" if torch.cuda.is_available() else "cpu")
 print(f"Is CUDA available: {torch.cuda.is_available()}")
-print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+if torch.cuda.is_available():
+    print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
 
 model = ControlModel(model, list(range(-5, -18, -1)))