Update app.py

app.py

@@ -53,30 +53,14 @@ class CodeLlamaTool(Tool):
             model=self.model,
             tokenizer=self.tokenizer,
             max_new_tokens=512,
-            temperature=0.2
+            truncation=True,
+            temperature=0.1            
         )
 
     def forward(self, question: str) -> str:
         prompt = f"""
-You are an advanced reasoning assistant. Use Python code if helpful.
-
-Instructions:
-- Solve step-by-step.
-- Wrap any code in <tool>...</tool>.
-- End your answer with <final>...</final> with the final result only.
-
-Example:
-Question: What is 3 times the square root of 49?
-Answer:
-<tool>
-import math
-print(3 * math.sqrt(49))
-</tool>
-<final>21.0</final>
-
-Now solve:
-Question: {question}
-Answer:
+    "You are an assistant. Solve step-by-step. "
+    "End your answer with a concise answer ina few words: FINAL ANSWER: [value]"
 """
 
         result = self.pipeline(prompt)[0]["generated_text"]