Update app.py

app.py

@@ -42,7 +42,7 @@ with gr.Blocks() as demo:
             top_p = gr.Slider(0.1, 1.0, value=0.9, step=0.05, label="Top-P (nucleus) sampling")
             rep_pty = gr.Slider(label="Repetition Penalty",info="Token repetition penalty. 1.0 means no penalty.",minimum=1.0, maximum=2.0, step=0.01, value=1.2)
             
-    generate_btn.click(generate_response, [instruction, max_new_tokens, temperature, top_k, top_p], output_box)
+    generate_btn.click(generate_response, [instruction, max_new_tokens, temperature, top_k, top_p, rep_pty], output_box)
 
 # ---- Launch ----
 if __name__ == "__main__":