Clear inputs on model change

app.py

@@ -46,9 +46,8 @@ def model_gen(content, model_type: str, chat_history):
     print(MODEL_LIB[model_type])
     return gen(content, chat_history)
 
-def clear_component(comp):
-    with gradio.Blocks():
-        clear = gradio.ClearButton()
+def clear_chat(chatbot, msg):
+    return '', []
 
 MODEL_LIB = {'RUBASE': giga_gen, 'TINYLLAMA': tiny_gen, 'QWEN2INS1B': qwen_gen}
 
@@ -57,11 +56,11 @@ 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(model_gen, [gradio.Textbox(), model_left, gradio.Chatbot()], [gradio.Textbox(), chatbot_left])
+            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(model_gen, [gradio.Textbox(), model_right, gradio.Chatbot()], [gradio.Textbox(), chatbot_right])
+            model_right.change(clear_chat, [chatbot_right, msg], [chatbot_right, msg])
 
     with gradio.Row():
         gradio.Button('Both Good')