Clear inputs on model change

app.py

@@ -47,7 +47,7 @@ def model_gen(content, model_type: str, chat_history):
     return gen(content, chat_history)
 
 def clear_chat(chatbot, msg):
-    return '', []
+    return []
 
 MODEL_LIB = {'RUBASE': giga_gen, 'TINYLLAMA': tiny_gen, 'QWEN2INS1B': qwen_gen}
 
@@ -56,11 +56,9 @@ def tab_arena():
         with gradio.Column():
             model_left = gradio.Dropdown(["TINYLLAMA", "QWEN2INS1B", "RUBASE"], value="TINYLLAMA", interactive=True, multiselect=False, label="Left model")
             chatbot_left = gradio.Chatbot()
-            model_left.change(clear_chat, [chatbot_left, msg], [chatbot_left, msg])
         with gradio.Column():
             model_right = gradio.Dropdown(["TINYLLAMA", "QWEN2INS1B", "RUBASE"], value= "RUBASE", interactive=True, multiselect=False, label="Right model")
             chatbot_right = gradio.Chatbot()
-            model_right.change(clear_chat, [chatbot_right, msg], [chatbot_right, msg])
 
     with gradio.Row():
         gradio.Button('Both Good')
@@ -71,8 +69,11 @@ def tab_arena():
     msg = gradio.Textbox()
     clear = gradio.ClearButton([msg, chatbot_left, chatbot_right])
     
-    msg.submit(model_gen, [msg, model_left, chatbot_left], [msg, chatbot_left])
-    msg.submit(model_gen, [msg, model_right, chatbot_right], [msg, chatbot_right])
+    with gradio.Blocks():
+        model_left.change(clear_chat, [chatbot_left, msg], [chatbot_left, msg])
+        model_right.change(clear_chat, [chatbot_right, msg], [chatbot_right, msg])
+        msg.submit(model_gen, [msg, model_left, chatbot_left], [msg, chatbot_left])
+        msg.submit(model_gen, [msg, model_right, chatbot_right], [msg, chatbot_right])
 
         # with gradio.Column():
         #     gradio.ChatInterface(