add also GPU options

app.py

@@ -8,7 +8,7 @@ tokenizer = AutoTokenizer.from_pretrained("deepseek-ai/deepseek-coder-1.3b-instr
 model = AutoModelForCausalLM.from_pretrained("deepseek-ai/deepseek-coder-1.3b-instruct", trust_remote_code=True, torch_dtype=torch.bfloat16)
 
 # Use CPU if CUDA is not available
-device = torch.device("cpu")
+device = torch.device("cpu" if not torch.cuda.is_available() else "cuda")
 model = model.to(device)
 
 # Theme builder