Spaces:

smjain
/

chat

Runtime error

smjain commited on Jul 21, 2022

Commit

6854f2b

1 Parent(s): b19af15

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ mdl = AutoModelForCausalLM.from_pretrained("facebook/blenderbot-400M-distill")
 def converse(user_input, chat_history=[]):
-    user_input_ids = chat_tkn(user_input + chat_tkn.eos_token, return_tensors='pt').input_ids
     # create a combined tensor with chat history
     bot_input_ids = torch.cat([torch.LongTensor(chat_history), user_input_ids], dim=-1)

 def converse(user_input, chat_history=[]):
+    user_input_ids = chat_tkn.encode(user_input + chat_tkn.eos_token, return_tensors='pt')
     # create a combined tensor with chat history
     bot_input_ids = torch.cat([torch.LongTensor(chat_history), user_input_ids], dim=-1)