Update app.py

app.py

@@ -53,6 +53,7 @@ mname = "facebook/blenderbot-400M-distill"
 model = BlenderbotForConditionalGeneration.from_pretrained(mname)
 tokenizer = BlenderbotTokenizer.from_pretrained(mname)
 
+
 def take_last_tokens(inputs, note_history, history):
     """Filter the last 128 tokens"""
     if inputs['input_ids'].shape[1] > 128:
@@ -61,6 +62,12 @@ def take_last_tokens(inputs, note_history, history):
         note_history = ['</s> <s>'.join(note_history[0].split('</s> <s>')[2:])]
         history = history[1:]
     return inputs, note_history, history
+    
+def add_note_to_history(note, note_history):
+    """Add a note to the historical information"""
+    note_history.append(note)
+    note_history = '</s> <s>'.join(note_history)
+    return [note_history]
 
 
 title = "💬ChatBack🧠💾"