Update app.py

app.py

@@ -54,7 +54,7 @@ def format_prompt(message, history):
 result = []
 
 
-def chat_inf(system_prompt,prompt,history,memory,client_choice,seed,temp,tokens,top_p,rep_p,chat_mem):
+def chat_inf(system_prompt,prompt,history,client_choice,seed,temp,tokens,top_p,rep_p,chat_mem):
     #token max=8192
     hist_len=0
     client=clients[int(client_choice)-1]
@@ -98,7 +98,7 @@ def chat_inf(system_prompt,prompt,history,memory,client_choice,seed,temp,tokens,
         yield [(prompt,output)]
     history.append((prompt,output))
     memory=history
-    yield history,memory
+    yield history
 
 def get_screenshot(chat: list,height=5000,width=600,chatblock=[],theme="light",wait=3000,header=True):
     print(chatblock)
@@ -168,7 +168,7 @@ with gr.Blocks() as app:
     im_go=im_btn.click(get_screenshot,[chat_b,im_height,im_width,chatblock,theme,wait_time],img)
     chat_sub=inp.submit(check_rand,[rand,seed],seed).then(chat_inf,[sys_inp,inp,chat_b,client_choice,seed,temp,tokens,top_p,rep_p,chat_mem],chat_b)
     
-    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],[chat_b,memory])
+    go=btn.click(check_rand,[rand,seed],seed).then(chat_inf,[sys_inp,inp,chat_b,client_choice,seed,temp,tokens,top_p,rep_p,chat_mem],[chat_b])
     
     stop_btn.click(None,None,None,cancels=[go,im_go,chat_sub])
     clear_btn.click(clear_fn,None,[inp,sys_inp,chat_b])