Update app.py

app.py

@@ -45,6 +45,7 @@ def run_generation(user_text, top_p, temperature, top_k, max_new_tokens):
         max_new_tokens=max_new_tokens,
         do_sample=True,
         top_p=top_p,
+        repetition_penalty=1.5,
         temperature=float(temperature),
         top_k=top_k,
     )
@@ -101,7 +102,7 @@ with gr.Blocks() as demo:
 
     user_text.submit(
         run_generation,
-        [user_text,repetition_penalty=1.5,temperature,do_sample=True,top_k, max_new_tokens],
+        [user_text,temperature,top_k, max_new_tokens],
         model_output,
     )
     button_submit.click(