nugentc commited on
Commit
a92a6ee
·
1 Parent(s): 91f2d69
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ def chat(message, history=[]):
29
  bot_input_ids = new_user_input_ids
30
  response_ids = model.generate(bot_input_ids, max_length=500, pad_token_id=tokenizer.eos_token_id)
31
  response = tokenizer.decode(response_ids).replace("<|endoftext|>", "")
32
- bot_input_ids = torch.cat([last_set_of_ids, response_ids], dim=-1)
33
  history.push((message, response, bot_input_ids))
34
  return response, history, feedback(message)
35
 
 
29
  bot_input_ids = new_user_input_ids
30
  response_ids = model.generate(bot_input_ids, max_length=500, pad_token_id=tokenizer.eos_token_id)
31
  response = tokenizer.decode(response_ids).replace("<|endoftext|>", "")
32
+ bot_input_ids = torch.cat([bot_input_ids, response_ids], dim=-1)
33
  history.push((message, response, bot_input_ids))
34
  return response, history, feedback(message)
35