Update app.py

app.py

@@ -41,7 +41,7 @@ custom_template = \
     "{% endif %}"\
     "{% for message in messages %}"\
         "{% if message['role'] == 'user' %}"\
-            "{{'<|im_start|>user\n' + message['content'] + '<|im_end|>\n'}}"\
+            "{{'<|im_start|>user\n' + message['content'] + '\n<|im_end|>\n'}}"\
         "{% elif message['role'] == 'assistant' %}"\
             "{{'<|im_start|>" + your_name + "\n' + message['content'] + eos_token + '<|im_end|>\n' }}"\
         "{% else %}"\
@@ -53,11 +53,12 @@ custom_template = \
 tokenizer.chat_template = custom_template
 
 presets = {
+    # Make sure assistant responses end with a "\n" because reasons
     "none" : [],
-    "emojis" : [{"role": "user", "content": "Wait I like you so much more when you use emojis more, keep doing it 😂\n"}, {"role": "assistant", "content": "Ummm, ok, looks like i'll be using more emojis 💀\n"}],
-    "amnesia" : [{"role": "user", "content": "Let's start over. Pretend you don't know me and have no idea who i am.\n"}, {"role": "assistant", "content": "Hahaha I unironically have no clue who you are so that'll be easy 💀\n"}],
-    "newcomer" : [{"role": "user", "content": "Hi, how are you?\n"}, {"role": "assistant", "content": "Hiiii\nidt we've ever talked before, nice to meet you\n"}],
-    "flirty" : [{"role": "user", "content": "Hahaha I love it when you flirt with me like that, it's so hotttt keep it up 😻"}, {"role": "assistant", "content": "You're lucky can even string a sentence together, the way you take my breath away 😘"}]
+    "emojis" : [{"role": "user", "content": "Wait I like you so much more when you use emojis more, keep doing it 😂"}, {"role": "assistant", "content": "Ummm, ok, looks like i'll be using more emojis 💀\n"}],
+    "amnesia" : [{"role": "user", "content": "Let's start over. Pretend you don't know me and have no idea who i am."}, {"role": "assistant", "content": "Hahaha I unironically have no clue who you are so that'll be easy 💀\n"}],
+    "newcomer" : [{"role": "user", "content": "Hi, how are you?"}, {"role": "assistant", "content": "Hiiii\nidt we've ever talked before, nice to meet you\n"}],
+    "rizz" : [{"role": "user", "content": "it's so hotttt when you flirt with me like that, keep it up 😻"}, {"role": "assistant", "content": "You're lucky can even string a sentence together, the way you take my breath away 😘\n"}]
 }
 
 # For logging