Clear inputs on model change

app.py

@@ -44,6 +44,10 @@ 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()
+
 
 MODEL_LIB = {'RUBASE': giga_gen, 'TINYLLAMA': tiny_gen, 'QWEN2INS1B': qwen_gen}
 
@@ -52,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(gradio.ClearButton, [chatbot_left], None)
+            model_left.change(lambda x: gradio.Chatbot(), [chatbot_left], [chatbot_left])
         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(gradio.ClearButton, [chatbot_right], None)
+            model_right.change(lambda x: gradio.Chatbot(), [chatbot_right], [chatbot_right])
 
     with gradio.Row():
         gradio.Button('Both Good')