winglian commited on
Commit
a874b38
1 Parent(s): b39d980

random sample deprecation replace with shuffle

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -133,7 +133,8 @@ def chat(history1, history2, system_msg):
133
  history1 = history1 or []
134
  history2 = history2 or []
135
 
136
- random_battle = random.sample(AVAILABLE_MODELS.keys(), 2)
 
137
  model1 = get_model_pipeline(random_battle[0])
138
  model2 = get_model_pipeline(random_battle[1])
139
 
 
133
  history1 = history1 or []
134
  history2 = history2 or []
135
 
136
+ arena_bots = list(AVAILABLE_MODELS.keys())
137
+ random_battle = arena_bots[0:2]
138
  model1 = get_model_pipeline(random_battle[0])
139
  model2 = get_model_pipeline(random_battle[1])
140