<usr>, <bot> 토큰이 아니라 0:, 1: 으로 수정

app.py

@@ -15,10 +15,11 @@ def query(message, chat_history, max_turn=4):
     if len(chat_history) > max_turn:
         chat_history = chat_history[-max_turn:]
     for user, bot in chat_history:
-        prompt.append(f"<usr> {user}")
-        prompt.append(f"<bot> {bot}")
-    prompt.append(f"<usr> {message}")
-    prompt = "\n".join(prompt) + "\n<bot>"
+        prompt.append(f"0 : {user}")
+        prompt.append(f"1 : {bot}")
+    
+    prompt.append(f"0 : {message}")
+    prompt = "\n".join(prompt) + "\n1 : "
 
     output = generator(
         prompt,