Update app.py

app.py

@@ -113,13 +113,17 @@ with gr.Blocks() as demo:
     history = gr.State(init_history)
 
     submitBtn.click(
-        predict, [user_input, max_length, top_p, temperature, history],
+        predict,
+        [user_input, max_length, top_p, temperature, history],
         [five_sample, history], show_progress=True
     )
     submitBtn.click(reset_user_input, [], [user_input])
     submitBtn.click(reset_user_input, [], [final_sample])
 
-    expand.select(fn=predict, inputs=expand, outputs=final_sample)
+    expand.select(
+        fn=predict,
+        inputs=[expand, max_length, top_p, temperature, history],
+        outputs=final_sample)
 
     emptyBtn.click(reset_state,
                    outputs=[five_sample, final_sample, history],