Update app.py

app.py

@@ -38,7 +38,7 @@ class StarlingBot:
             if mode == "Assistant":
                 conversation = f" GPT4 Correct Assistant: {system_prompt if system_prompt else self.system_prompt}  GPT4 Correct Assistant: {assistant_message if assistant_message else ''}  GPT4 Correct User: {user_message}  GPT4 Correct Assistant:"
             else:  # mode == "Coder"
-                conversation = f" Code Assistant: {system_prompt if system_prompt else self.system_prompt}  Code Assistant: {assistant_message if assistant_message else ''}  GPT4 Correct User: {user_message}  Code Assistant:"
+                conversation = f" Code Assistant: {system_prompt if system_prompt else self.system_prompt}  Code Assistant: {assistant_message if assistant_message else ''}  Code User:: {user_message}  Code Assistant:"
             input_ids = tokenizer.encode(conversation, return_tensors="pt", add_special_tokens=True)
             input_ids = input_ids.to(device)
             response = model.generate(