Update app.py

app.py

@@ -101,6 +101,9 @@ def check_rand(inp, val):
     else:
         return gr.Slider(label="Seed", minimum=1, maximum=1111111111111111, value=int(val))
 
+def 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):
+    return chat_inf(sys_inp, inp, chat_b, memory, client_choice, seed, temp, tokens, top_p, rep_p, chat_mem, custom_prompt, translate_to_persian_checkbox)
+
 with gr.Blocks() as app:
     memory = gr.State()
     chat_b = gr.Chatbot(height=500)
@@ -133,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_inf, [sys_inp, inp, chat_b, memory, client_choice, seed, temp, tokens, top_p, rep_p, chat_mem, custom_prompt, translate_to_persian_checkbox.value], [chat_b, memory])
-    go = btn.click(check_rand, [rand, seed], seed).then(chat_inf, [sys_inp, inp, chat_b, memory, client_choice, seed, temp, tokens, top_p, rep_p, chat_mem, custom_prompt, translate_to_persian_checkbox.value], [chat_b, 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(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)
 
     clear_btn.click(clear_fn, None, [inp, sys_inp, chat_b, memory])
 app.queue(default_concurrency_limit=10).launch()