Clear inputs on model change

app.py

@@ -39,7 +39,7 @@ def qwen_gen(content, chat_history):
     send_to_s3(res, f'protobench/tiny_{str(datetime.now()).replace(" ", "_")}.json', S3_SESSION)
     return '', chat_history
 
-def model_gen(content, model_type: str, chat_history):
+def model_gen(content, chat_history, model_type: str):
     if content is None:
         return None, None
     gen = MODEL_LIB[model_type]
@@ -49,8 +49,12 @@ def model_gen(content, model_type: str, chat_history):
 def clear_chat():
     return '', []
 
-def rerun(msg, chatbot):
-    return msg, chatbot[:-1]
+def model_regen(content, chat_history, model_type: str):
+    if content is None:
+        return None, None
+    gen = MODEL_LIB[model_type]
+    print(MODEL_LIB[model_type])
+    return gen(content, chat_history[:-1])
 
 MODEL_LIB = {'RUBASE': giga_gen, 'TINYLLAMA': tiny_gen, 'QWEN2INS1B': qwen_gen}
 
@@ -72,9 +76,9 @@ def tab_arena():
     with gradio.Row():
         with gradio.Accordion("Parameters", open=False):
             context = gradio.Checkbox(label="No context", value=False)
-            top_p = gradio.Slider(label='Top P', minimum=0, maximum=1, value=1, step=0.5)
-            temp = gradio.Slider(label='Temperature', minimum=0, maximum=1, value=0.7, step=0.5)
-            max_tokens = gradio.Slider(label='Max ouput tokens', minimum=1, maximum=2048, value=512, step=1)
+            top_p = gradio.Slider(label='Top P', minimum=0, maximum=1, value=1, step=0.5, interactive=True)
+            temp = gradio.Slider(label='Temperature', minimum=0, maximum=1, value=0.7, step=0.5, interactive=True)
+            max_tokens = gradio.Slider(label='Max ouput tokens', minimum=1, maximum=2048, value=512, step=1, interactive=True)
 
     with gradio.Row():
         msg = gradio.Textbox()
@@ -83,14 +87,14 @@ def tab_arena():
         clear = gradio.ClearButton([msg, chatbot_left, chatbot_right], value='Clear history')
         regen_left = gradio.Button(value='Regenerate left answer')
         regen_right = gradio.Button(value='Regenerate right answer')
-        regen_left.click(rerun, [msg, chatbot_left], [msg, chatbot_left])
-        regen_right.click(rerun, [msg, chatbot_right], [msg, chatbot_left])
+        regen_left.click(model_regen, [msg, chatbot_left, model_left], [msg, chatbot_left])
+        regen_right.click(model_regen, [msg, chatbot_right, model_right], [msg, chatbot_left])
     
     with gradio.Blocks():
         model_left.change(clear_chat, [], [msg, chatbot_left])
         model_right.change(clear_chat, [], [msg, 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])
+        msg.submit(model_gen, [msg, chatbot_left, model_left], [msg, chatbot_left])
+        msg.submit(model_gen, [msg, chatbot_right, model_right], [msg, chatbot_right])
 
         # with gradio.Column():
         #     gradio.ChatInterface(