Update app.py

app.py

@@ -52,7 +52,7 @@ Example format:
 <count> [remaining budget] </count>
 <answer> [Final Answer] </answer> (must give final answer in this format)
 <reflection> [Evaluation of the solution] </reflection>
-<reward> [Float between 0.0 and 1.0] </reward> [/INST] [INST] [QUERY] {query}"""
+<reward> [Float between 0.0 and 1.0] </reward> [/INST] [INST] [QUERY] {query} [/INST] [ASSISTANT] """
 
     stream = client.text_generation(message, max_new_tokens=4096, stream=True, details=True, return_full_text=False)
     output = ""