Update app.py

app.py

@@ -136,8 +136,8 @@ with gr.Blocks() as app:
     client_choice.change(load_models, client_choice, [chat_b])
     app.load(load_models, client_choice, [chat_b])
 
-    chat_sub = inp.submit(check_rand, [rand, seed], seed).then(chat_wrapper, [sys_inp, inp, chat_b, memory, client_choice, seed, temp, tokens, top_p, rep_p, chat_mem, custom_prompt, translate_to_persian_checkbox]).then(display=chat_b, state=memory)
-    go = btn.click(check_rand, [rand, seed], seed).then(chat_wrapper, [sys_inp, inp, chat_b, memory, client_choice, seed, temp, tokens, top_p, rep_p, chat_mem, custom_prompt, translate_to_persian_checkbox]).then(display=chat_b, state=memory)
+    chat_sub = inp.submit(check_rand, [rand, seed], seed).then(chat_wrapper, [sys_inp, inp, chat_b, memory, client_choice, seed, temp, tokens, top_p, rep_p, chat_mem, custom_prompt, translate_to_persian_checkbox]).then(chat_b.display, memory)
+    go = btn.click(check_rand, [rand, seed], seed).then(chat_wrapper, [sys_inp, inp, chat_b, memory, client_choice, seed, temp, tokens, top_p, rep_p, chat_mem, custom_prompt, translate_to_persian_checkbox]).then(chat_b.display, memory)
 
     clear_btn.click(clear_fn, None, [inp, sys_inp, chat_b, memory])
 app.queue(default_concurrency_limit=10).launch()