Fix: GPU usage

app.py

@@ -130,7 +130,7 @@ def load_model():
     return model, device
 
 # Text generation function
-@spaces.gpu(enable_queue=True)
+@spaces.GPU(enable_queue=True)
 def generate_text(prompt, num_tokens, temperature=0.8):
     enc = tiktoken.get_encoding('gpt2')
     x = torch.tensor([enc.encode(prompt)], dtype=torch.long, device=device)