teaevo commited on
Commit
3bd433e
·
1 Parent(s): 3933abd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -66,7 +66,7 @@ def chatbot_response(user_message):
66
  # generated a response while limiting the total chat history to 1000 tokens,
67
  chat_history_ids = chatbot_model.generate(bot_input_ids, max_length=1000, pad_token_id=chatbot_tokenizer.eos_token_id)
68
 
69
- response = tokenizer.decode(chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens=True)
70
 
71
  new_chat = False
72
 
 
66
  # generated a response while limiting the total chat history to 1000 tokens,
67
  chat_history_ids = chatbot_model.generate(bot_input_ids, max_length=1000, pad_token_id=chatbot_tokenizer.eos_token_id)
68
 
69
+ response = chatbot_tokenizer.decode(chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens=True)
70
 
71
  new_chat = False
72