fix: add more GPU decorators

model.py

@@ -104,6 +104,7 @@ def get_prompt(messages, criteria_name):
     )
 
 
+@spaces.GPU
 def generate_tokens(prompt):
     result = model.generate(
         prompt=[prompt],