Update app.py

app.py

@@ -27,7 +27,7 @@ LINEBREAK_TOKEN = 13
 
 ROLE_TOKENS = {
     "user": USER_TOKEN,
-    "assistant": BOT_TOKEN,
+    "bot": BOT_TOKEN,
     "system": SYSTEM_TOKEN
 }
 
@@ -77,7 +77,7 @@ def bot(
         message_tokens = get_message_tokens(model=model, role="user", content=user_message)
         tokens.extend(message_tokens)
         if bot_message:
-            message_tokens = get_message_tokens(model=model, role="assistant", content=bot_message)
+            message_tokens = get_message_tokens(model=model, role="bot", content=bot_message)
             tokens.extend(message_tokens)
 
     last_user_message = history[-1][0]