Update app.py

app.py

@@ -7,14 +7,14 @@ chat_tkn = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-400M-distill
 mdl = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-400M-distill")
 
 def converse(user_input, chat_history=[]):
-    #tkn_ids = chat_tkn(input(">> User:") + chat_tkn.eos_token, return_tensors='pt')
+    
     user_input_ids = chat_tkn(user_input + chat_tkn.eos_token, return_tensors='pt')
 
-    # append the new user input tokens to the chat history
-    bot_input_ids = torch.cat([torch.LongTensor(history), user_input_ids], dim=-1)
+    # create a combined tensor with chat history
+    bot_input_ids = torch.cat([torch.LongTensor(chat_history), user_input_ids], dim=-1)
 
     # generate a response 
-    history = mdl.generate(bot_input_ids, max_length=1000, pad_token_id=chat_tkn.eos_token_id).tolist()
+    chat_history = mdl.generate(bot_input_ids, max_length=1000, pad_token_id=chat_tkn.eos_token_id).tolist()
 
     # convert the tokens to text, and then split the responses into lines
     response = chat_tkn.decode(chat_history[0]).split("<|endoftext|>")
@@ -27,7 +27,7 @@ def converse(user_input, chat_history=[]):
         html += "<div class='msg {}'> {}</div>".format(cls, msg)
     html += "</div>"
     
-    return html, history
+    return html, chat_history
 
 import gradio as gr