KvrParaskevi commited on
Commit
2cc7b13
1 Parent(s): 6704e38

Update chatbot.py

Browse files
Files changed (1) hide show
  1. chatbot.py +2 -2
chatbot.py CHANGED
@@ -66,7 +66,7 @@ llm = load_pipeline()
66
 
67
  def demo_miny_memory():
68
  #prompt = ChatPromptTemplate.from_template(template)
69
- memory = ConversationBufferMemory(llm = llm)
70
  return memory
71
 
72
  def demo_chain(input_text,history):
@@ -80,5 +80,5 @@ def demo_chain(input_text,history):
80
  memory=demo_miny_memory()
81
  )
82
 
83
- chat_reply = conversation.predict(input=input_text, return_only_outputs=True)
84
  return chat_reply #['response'].split('AI:')[-1]
 
66
 
67
  def demo_miny_memory():
68
  #prompt = ChatPromptTemplate.from_template(template)
69
+ memory = ConversationBufferMemory(llm = llm, memory_key = "history")
70
  return memory
71
 
72
  def demo_chain(input_text,history):
 
80
  memory=demo_miny_memory()
81
  )
82
 
83
+ chat_reply = conversation.invoke(input=input_text, return_only_outputs=True)
84
  return chat_reply #['response'].split('AI:')[-1]