nugentc commited on
Commit
d7c2aca
·
1 Parent(s): 30a909b
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ def chat(message, history=[]):
25
  bot_input_ids = torch.cat([torch.LongTensor(history), new_user_input_ids], dim=-1)
26
  history = model.generate(bot_input_ids, max_length=500, pad_token_id=tokenizer.eos_token_id).tolist()
27
  response = tokenizer.decode(history[0]).replace("<|endoftext|>", "")
28
- return response, history, 'haha'
29
 
30
 
31
  def feedback(text):
 
25
  bot_input_ids = torch.cat([torch.LongTensor(history), new_user_input_ids], dim=-1)
26
  history = model.generate(bot_input_ids, max_length=500, pad_token_id=tokenizer.eos_token_id).tolist()
27
  response = tokenizer.decode(history[0]).replace("<|endoftext|>", "")
28
+ return history, history, 'haha'
29
 
30
 
31
  def feedback(text):