Update app.py

app.py

@@ -107,12 +107,12 @@ def get_response(system_message, chat_history, user_text, max_new_tokens=256, co
         prompt = st.session_state.last_response
     else:
         # Build the conversation prompt
-        prompt = "" 
+        prompt = "<|assistant|>"+system_message+f"<|end|>\n"
         # f"{system_message}\nCurrent Conversation:\n"
         for message in chat_history:
             role = "<|assistant|>" if message['role'] == 'assistant' else "<|user|>"
-            prompt += f"\n{role}\n{message['content']}\n<|user|>\n"        
-        prompt += f"\n<|user|>\n{user_text}\n<|assistant|>\n"
+            prompt += f"\n{role}\n{message['content']}\n<|end|>\n"        
+        prompt += f"\n<|user|>\n{user_text}\n<|end|><|assistant|>\n"
 
     # Generate the response
     response_output = generator(