Spaces:

openaccess-ai-collective
/

rlhf-arena

Runtime error

winglian commited on Jun 4, 2023

Commit

a874b38

1 Parent(s): b39d980

random sample deprecation replace with shuffle

Files changed (1) hide show

app.py CHANGED Viewed

@@ -133,7 +133,8 @@ def chat(history1, history2, system_msg):
     history1 = history1 or []
     history2 = history2 or []
-    random_battle = random.sample(AVAILABLE_MODELS.keys(), 2)
     model1 = get_model_pipeline(random_battle[0])
     model2 = get_model_pipeline(random_battle[1])

     history1 = history1 or []
     history2 = history2 or []
+    arena_bots = list(AVAILABLE_MODELS.keys())
+    random_battle = arena_bots[0:2]
     model1 = get_model_pipeline(random_battle[0])
     model2 = get_model_pipeline(random_battle[1])