Update app.py

app.py

@@ -35,7 +35,7 @@ class TuluChatBot:
         self.system_message = new_system_message
 
     def format_prompt(self, user_message):
-        prompt = f"<|assistant|>\n {self.system_message}\n\n <|user|>{user_message}\n\n<|assistant|>\n"
+        prompt = f"<|assistant|>\n{self.system_message}\n<|user|>{user_message}\n<|assistant|>\n"
         return prompt
 
     def Tulu(self, user_message, temperature, max_new_tokens, top_p, repetition_penalty, do_sample):
@@ -57,6 +57,7 @@ class TuluChatBot:
 
         response = self.tokenizer.decode(output_ids[0], skip_special_tokens=True)
         response = response.strip()
+        response = response.split("<|assistant|>\n")[-1]
         return response
 
 def gradio_Tulu(user_message, system_message, max_new_tokens, temperature, top_p, repetition_penalty, do_sample):