Update app.py

app.py

@@ -26,7 +26,7 @@ client = Client(
 )
 
 
-def generate(prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,  do_save=True):
+def generate(prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0):
 
     temperature = float(temperature)
     if temperature < 1e-2:
@@ -129,7 +129,7 @@ with gr.Blocks(theme=theme, analytics_enabled=False, css=css) as demo:
                     info="Penalize repeated tokens",
                 )
 
-    submit.click(generate, inputs=[instruction, temperature, max_new_tokens, top_p, repetition_penalty, do_save], outputs=[output])
+    submit.click(generate, inputs=[instruction, temperature, max_new_tokens, top_p, repetition_penalty], outputs=[output])
     instruction.submit(generate, inputs=[instruction, temperature, max_new_tokens, top_p, repetition_penalty], outputs=[output])
 
 demo.queue(concurrency_count=16).launch(debug=True)