winglian commited on
Commit
f9128f4
1 Parent(s): c395137

fix indexing

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -163,10 +163,10 @@ def chat(history1, history2, system_msg):
163
  len_tokens_model2 = len(tokens_model2)
164
  max_tokens = max(len_tokens_model1, len_tokens_model2)
165
  for i in range(0, max_tokens):
166
- if i <= len_tokens_model1:
167
  answer1 = tokens_model1[i]
168
  history1[-1][1] += answer1
169
- if i <= len_tokens_model2:
170
  answer2 = tokens_model2[i]
171
  history2[-1][1] += answer2
172
  # stream the response
 
163
  len_tokens_model2 = len(tokens_model2)
164
  max_tokens = max(len_tokens_model1, len_tokens_model2)
165
  for i in range(0, max_tokens):
166
+ if i < len_tokens_model1:
167
  answer1 = tokens_model1[i]
168
  history1[-1][1] += answer1
169
+ if i < len_tokens_model2:
170
  answer2 = tokens_model2[i]
171
  history2[-1][1] += answer2
172
  # stream the response