add cuda check

app.py

@@ -12,6 +12,10 @@ import transformers
 from huggingface_hub import CommitScheduler, hf_hub_download, login
 from transformers import AutoTokenizer
 
+print(f"Is CUDA available: {torch.cuda.is_available()}")
+
+print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+
 HF_TOKEN = os.getenv("HF_TOKEN")
 login(HF_TOKEN)