Clear inputs on model change

app.py

@@ -40,6 +40,8 @@ def qwen_gen(content, chat_history):
     return '', chat_history
 
 def model_gen(content, model_type: str, chat_history):
+    if content is None:
+        return None, None
     gen = MODEL_LIB[model_type]
     print(MODEL_LIB[model_type])
     return gen(content, chat_history)
@@ -56,11 +58,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(lambda x: gradio.Chatbot(), [chatbot_left], [chatbot_left])
+            model_left.change(model_gen, [None, model_left, None], [None, None])
         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(lambda x: gradio.Chatbot(), [chatbot_right], [chatbot_right])
+            model_right.change(model_gen, [None, model_right, None], [None, None])
 
     with gradio.Row():
         gradio.Button('Both Good')