Update app.py

app.py

@@ -54,7 +54,9 @@ tokenizer.chat_template = custom_template
 
 presets = {
     "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"}],
-    "none" : []
+    "none" : [],
+    "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!\n I don't think we've ever talked before, nice to meet you\n"}],
 }
 
 
@@ -66,9 +68,9 @@ def respond(
     mirostat_tau,
     mirostat_eta,
     frequency_penalty,
-    presence_penalty
+    presence_penalty,
+    preset
 ):
-    preset = "none"
     
     messages = presets[preset].copy()
     for val in history:
@@ -110,14 +112,15 @@ demo = gr.ChatInterface(
     description="The model may take a while if it hasn't run recently or a lot of people are using it",
     title="EliGPT v1.idon'tfuckingknow",
     additional_inputs=[
-        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
-        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
+        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens", info="How many words can the model generate?"),
+        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, sep=0.1, label="Temperature", info="How chaotic should the model be?"),
         gr.Slider(
             minimum=0.0,
             maximum=10.0,
             value=3.0,
             step=0.5,
             label="Mirostat tau",
+            info="Basically, how many drugs should the model be on?"
         ),
         gr.Slider(
             minimum=0.0,
@@ -125,6 +128,7 @@ demo = gr.ChatInterface(
             value=0.1,
             step=0.01,
             label="Mirostat eta",
+            info="I don't even know man"
         ),
         gr.Slider(
             minimum=0.0,
@@ -132,6 +136,7 @@ demo = gr.ChatInterface(
             value=0.1,
             step=0.01,
             label="Frequency penalty",
+            info='"Don\'repeat yourself"'
         ),
         gr.Slider(
             minimum=0.0,
@@ -139,7 +144,9 @@ demo = gr.ChatInterface(
             value=0.1,
             step=0.01,
             label="Presence penalty",
+            info='"Use lots of diverse words"'
         ),
+        gr.Radio(presets.keys(), label="Preset", info="Gaslight the model into acting a certain way")
     ],
 )