API fix

app.py

@@ -41,8 +41,10 @@ MODEL_LIB = {'RUBASE': giga_gen, 'TINYLLAMA': tiny_gen, 'QWEN2INS1B': qwen_gen}
 def tab_arena():
     with gradio.Row():
         with gradio.Column():
+            model_left = gradio.Dropdown(["TINYLLAMA", "QWEN2INS1B", "RUBASE"], value="TINYLLAMA", interactive=True, multiselect=False, label="Left model")
             chatbot_left = gradio.Chatbot()
         with gradio.Column():
+            model_right = gradio.Dropdown(["TINYLLAMA", "QWEN2INS1B", "RUBASE"], value= "RUBASE", interactive=True, multiselect=False, label="Right model")
             chatbot_right = gradio.Chatbot()
 
     with gradio.Row():
@@ -60,8 +62,10 @@ def tab_arena():
     #     time.sleep(2)
     #     return "", chat_history
 
-    msg.submit(MODEL_LIB[gradio.Dropdown(["TINYLLAMA", "QWEN2INS1B", "RUBASE"], multiselect=False, label="Left model")], [msg, chatbot_left], [msg, chatbot_left])
-    msg.submit(MODEL_LIB[gradio.Dropdown(["TINYLLAMA", "QWEN2INS1B", "RUBASE"], multiselect=False, label="Left model")], [msg, chatbot_right], [msg, chatbot_right])
+    model_right.submit()
+
+    msg.submit(giga_gen, [msg, chatbot_left], [msg, chatbot_left])
+    msg.submit(qwen_gen, [msg, chatbot_right], [msg, chatbot_right])
 
         # with gradio.Column():
         #     gradio.ChatInterface(