Update app.py

app.py

@@ -122,10 +122,10 @@ with gr.Blocks(title=title) as demo:
     
     with gr.Tab("Chat mode"):
         with gr.Row():
-            chatbot = gr.Chatbot()
             with gr.Column():
+                chatbot = gr.Chatbot()
                 msg = gr.Textbox(scale=4, show_label=False, placeholder="Enter text and press enter", container=False)
-                clear = gr.Button("Clear")
+                clear = gr.ClearButton([msg, chatbot])
             with gr.Column():
                 token_count_chat = gr.Slider(10, 512, label="Max Tokens", step=10, value=333)
                 temperature_chat = gr.Slider(0.2, 2.0, label="Temperature", step=0.1, value=1.2)
@@ -160,7 +160,6 @@ with gr.Blocks(title=title) as demo:
             msg.submit(user_msg, [msg, chatbot], [msg, chatbot], queue=False).then(
                 chat, chatbot, chatbot, api_name="chat"
             )
-            clear.click(clear_chat, None, [chatbot], queue=False)
             
     with gr.Tab("Instruct mode"):
         gr.Markdown(f"100% RNN RWKV-LM **trained on 100+ natural languages**. Demo limited to ctxlen {ctx_limit}. For best results, <b>keep your prompt short and clear</b>.")