davidberenstein1957 commited on
Commit
ea1ca1e
·
verified ·
1 Parent(s): 26647e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -44,8 +44,8 @@ def next_input(_prompt, _completion_a, _completion_b):
44
  row = new_dataset[0]
45
  messages = row["messages"][:-1]
46
  completions = [row["response"]]
47
- completions.append(_get_response(messages))
48
- completions.append(_get_response(messages))
49
  random.shuffle(completions)
50
  return messages, completions.pop(), completions.pop()
51
 
 
44
  row = new_dataset[0]
45
  messages = row["messages"][:-1]
46
  completions = [row["response"]]
47
+ completions.append(get_response(messages))
48
+ completions.append(get_response(messages))
49
  random.shuffle(completions)
50
  return messages, completions.pop(), completions.pop()
51