API fix

app.py

@@ -13,6 +13,9 @@ MERA_table = None
 TINYLLAMA = None
 QWEN2INS1B = None
 
+RIGHT_MODEL = None
+LEFT_MODEL = None
+
 S3_SESSION = None
 
 def giga_gen(content, chat_history):
@@ -36,6 +39,9 @@ def qwen_gen(content, chat_history):
     send_to_s3(res, f'protobench/tiny_{str(datetime.now()).replace(" ", "_")}.json', S3_SESSION)
     return '', chat_history
 
+def change_model(model, model_key):
+    model = MODEL_LIB[model_key]
+
 MODEL_LIB = {'RUBASE': giga_gen, 'TINYLLAMA': tiny_gen, 'QWEN2INS1B': qwen_gen}
 
 def tab_arena():
@@ -62,6 +68,12 @@ def tab_arena():
     #     time.sleep(2)
     #     return "", chat_history
 
+    model_left.change(change_model, [LEFT_MODEL, model_left])
+    model_right.change(change_model, [RIGHT_MODEL, model_right])
+
+    print(LEFT_MODEL)
+    print(RIGHT_MODEL)
+
     msg.submit(giga_gen, [msg, chatbot_left], [msg, chatbot_left])
     msg.submit(qwen_gen, [msg, chatbot_right], [msg, chatbot_right])