Update app.py

app.py

@@ -65,6 +65,8 @@ def respond(
     temperature,
     mirostat_tau,
     mirostat_eta,
+    frequency_penalty,
+    presence_penalty
 ):
     preset = "none"
     
@@ -89,7 +91,9 @@ def respond(
         mirostat_mode=1,
         mirostat_tau=mirostat_tau,
         mirostat_eta=mirostat_eta,
-        max_tokens=128
+        max_tokens=128,
+        frequency_penalty=frequency_penalty,
+        presence_penalty=presence_penalty,
     ):
         token = message["choices"][0]["text"]
 
@@ -122,6 +126,20 @@ demo = gr.ChatInterface(
             step=0.01,
             label="Mirostat eta",
         ),
+        gr.Slider(
+            minimum=0.0,
+            maximum=1.0,
+            value=0.1,
+            step=0.01,
+            label="Frequency penalty",
+        ),
+        gr.Slider(
+            minimum=0.0,
+            maximum=1.0,
+            value=0.1,
+            step=0.01,
+            label="Presence penalty",
+        ),
     ],
 )