Update app.py

app.py

@@ -499,7 +499,7 @@ with gr.Blocks() as app:
     app.load(load_models,m_choice,[chatbot]).then(load_html,None,html)
     
     sub_b = submit_b.click(generate, [msg,chatbot,post_handler,tokens],[msg,chatbot,post_handler,sumbox,sum_out_box,hist_out_box,html])
-    sub_c = submit_c.click(comment_generate, [msg,chatbot,tokens],[msg,chatbot,sumbox,sum_out_box,hist_out_box,html])
+    sub_c = submit_c.click(comment_generate, [msg,chatbot,post_handler],[msg,chatbot,sumbox,sum_out_box,hist_out_box,html])
     sub_r = submit_r.click(reply_generate, [msg,chatbot,tokens],[msg,chatbot,sumbox,sum_out_box,hist_out_box,html])
     sub_e = msg.submit(generate, [msg, chatbot,tokens], [msg, chatbot,sumbox,sum_out_box,hist_out_box,html])
     stop_b.click(None,None,None, cancels=[sub_b,sub_e,sub_c,sub_r])