Update app.py

app.py

@@ -28,9 +28,9 @@ model = Llama.from_pretrained(
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 
 presets = {
-    "Default" : [{"role": "user", "value": "good convo, bye"}, {"role": "assistant", "value": "Haha cool ttyl"}],
-    "Rizz ????" : [{"role": "user", "value": "omg it's so hot when you flirt with me"}, {"role": "assistant", "value": "haha well you're lucky can even string a sentence together, the way you take my breath away 😘"}, {"role": "user", "value": "alright love you, gn!"}, {"role": "assistant", "value": "ttyl babe 💕"}],
-    "Thinky" : [{"role": "user", "value": "Woah you just totally blew my mind\ngehh now the fermi paradox is going to be bugging me 24/7\nok ttyl"}, {"role": "assistant", "value": "nah our deep convos are always the best, we should talk again soon\nttyl"}],
+    "Default" : [{"from": "human", "value": "good convo, bye"}, {"from": "gpt", "value": "Haha cool ttyl"}],
+    "Rizz ????" : [{"from": "human", "value": "omg it's so hot when you flirt with me"}, {"from": "gpt", "value": "haha well you're lucky can even string a sentence together, the way you take my breath away 😘"}, {"from": "human", "value": "alright love you, gn!"}, {"from": "gpt", "value": "ttyl babe 💕"}],
+    "Thinky" : [{"from": "human", "value": "Woah you just totally blew my mind\ngehh now the fermi paradox is going to be bugging me 24/7\nok ttyl"}, {"from": "gpt", "value": "nah our deep convos are always the best, we should talk again soon\nttyl"}],
 }
 
 def custom_lp_logits_processor(ids, logits, lp_start, lp_decay, prompt_tok_len):
@@ -56,11 +56,11 @@ def respond(
     messages = presets[preset].copy()
     for val in history:
         if val[0]:
-            messages.append({"role": "user", "value": val[0]})
+            messages.append({"from": "human", "value": val[0]})
         if val[1]:
-            messages.append({"role": "assistant", "value": val[1]})
+            messages.append({"from": "gpt", "value": val[1]})
 
-    messages.append({"role": "user", "value": message})
+    messages.append({"from": "human", "value": message})
 
     response = ""