Update main.py

main.py

@@ -18,11 +18,11 @@ class Item(BaseModel):
     repetition_penalty: float = 1.0
 
 def format_prompt(message, history):
-    prompt = "<s>"
+    prompt = "<s> "
     for user_prompt, bot_response in history:
         prompt += f"[INST] {user_prompt} [/INST]"
-        prompt += f" {bot_response}</s> "
-    prompt += f"[INST] {message} [/INST]"
+        prompt += f" {bot_response} "
+    prompt += f" </s> [INST] {message} [/INST]"
     return prompt
 
 def generate(item: Item):