Check for GPU

app.py

@@ -8,6 +8,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 import os
 import spaces
 
+print("CUDA available:", torch.cuda.is_available())
+
 @spaces.GPU
 def claim_gpu():
     # Dummy function to make Spaces detect GPU usage