Update app.py

app.py

@@ -82,18 +82,17 @@ def format_prompt(state, bot_index, character_name, character_description, user_
         user_name = "You"
     
     prompt = f"The following is a conversation between {user_name} and {character_name}.\n\n"
-    prompt += f"{character_name}'s background:\n{character_description}\n\n"
     
     # Get the last num_messages messages from the conversation history
     recent_messages = state["history"][bot_index][-num_messages:]
     
     for message in recent_messages:
         if message['role'] == 'user':
-            prompt += f"{user_name}: {message['content']}\n"
+            prompt += f"<|im_start|>user {message['content']}<|im_end|>\n"
         else:
-            prompt += f"{character_name}: {message['content']}\n"
+            prompt += f"<|im_start|>assistant {message['content']}<|im_end|>\n"
     
-    prompt += f"{character_name}:"
+    prompt += f"<|im_start|>assistant"
     return prompt
 import aiohttp
 import asyncio