Update app.py

app.py

@@ -15,20 +15,20 @@ def format_prompt(message, history):
   return prompt
 
 def generate(
-    prompt, history, max_tokens=2000,
+    prompt, history, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
 ):
-    '''temperature = float(temperature)
+    temperature = float(temperature)
     if temperature < 1e-2:
         temperature = 1e-2
-    top_p = float(top_p)'''
+    top_p = float(top_p)
 
     generate_kwargs = dict(
-        #temperature=temperature,
-        max_tokens=max_tokens,
-        #top_p=top_p,
-        #repetition_penalty=repetition_penalty,
-        #do_sample=True,
-        #seed=42,
+        temperature=temperature,
+        max_new_tokens=max_new_tokens,
+        top_p=top_p,
+        repetition_penalty=repetition_penalty,
+        do_sample=True,
+        seed=42,
     )
 
     formatted_prompt = format_prompt(prompt, history)
@@ -44,15 +44,41 @@ def generate(
 
 additional_inputs=[
     gr.Slider(
-        label="Max tokens",
-        value=2000,
+        label="Temperature",
+        value=0.9,
+        minimum=0.0,
+        maximum=1.0,
+        step=0.05,
+        interactive=True,
+        info="Higher values produce more diverse outputs",
+    ),
+    gr.Slider(
+        label="Max new tokens",
+        value=256,
         minimum=0,
-        maximum=2048,
+        maximum=1048,
         step=64,
         interactive=True,
         info="The maximum numbers of new tokens",
     ),
-
+    gr.Slider(
+        label="Top-p (nucleus sampling)",
+        value=0.90,
+        minimum=0.0,
+        maximum=1,
+        step=0.05,
+        interactive=True,
+        info="Higher values sample more low-probability tokens",
+    ),
+    gr.Slider(
+        label="Repetition penalty",
+        value=1.2,
+        minimum=1.0,
+        maximum=2.0,
+        step=0.05,
+        interactive=True,
+        info="Penalize repeated tokens",
+    )
 ]