Update app.py

app.py

@@ -45,7 +45,6 @@ Few sentences of reasoning
 """
 
 # --- Helper Functions ---
-@spaces.GPU
 def format_rules(rules):
     formatted_rules = "<rules>\n"
     for i, rule in enumerate(rules):
@@ -142,7 +141,8 @@ class ModelWrapper:
                     message, tokenize=False, continue_final_message=True, enable_thinking=False
                 )
         return prompt
-
+    
+    @spaces.GPU
     def get_response(self, input, temperature=0.7, top_k=20, top_p=0.8, max_new_tokens=256,
                      enable_thinking=True, system_prompt=SYSTEM_PROMPT):
         print("Generating response...")