Update app.py

app.py

@@ -15,7 +15,7 @@ def format_prompt(message, history):
   return prompt
 
 def generate(
-    prompt, history, system_prompt, temperature=0.9, max_new_tokens=1000, top_p=0.95, repetition_penalty=1.0, seed=42
+    prompt, system_prompt, history, temperature, max_new_tokens, top_p, top_k, repetition_penalty, seed
 ):
     temperature = float(temperature)
     if temperature < 1e-2:
@@ -26,6 +26,7 @@ def generate(
         temperature=temperature,
         max_new_tokens=max_new_tokens,
         top_p=top_p,
+        top_k=top_k,
         repetition_penalty=repetition_penalty,
         do_sample=True,
         seed=seed,