Update app.py

app.py

@@ -96,13 +96,13 @@ def get_response(system_message, chat_history, user_text, max_new_tokens=256, co
     for message in chat_history:
         role = "<|assistant|>" if message['role'] == 'assistant' else "<|user|>"
         prompt += f"\n{role}\n{message['content']}\n"
-    prompt += f"\n<|user|>\n{user_text}\n<|assistant|>\n"
+    
 
     if continue_last:
         # We want to continue the last assistant response
-        prompt += "Assistant:"
+        # prompt += ""
     else:
-        prompt += f"User: {user_text}\nAssistant:"
+        prompt += f"\n<|user|>\n{user_text}\n<|assistant|>\n"
 
     # Generate the response
     response_output = generator(
@@ -150,9 +150,6 @@ with chat_interface:
     # User input area (moved to the bottom)
     st.session_state.user_text = st.chat_input(placeholder="Enter your text here.")
 
-# User input area (moved to the bottom)
-st.session_state.user_text = st.chat_input(placeholder="Enter your text here.")
-
 # When the user enters new text
 if st.session_state.user_text:
     # Display the user's message