KvrParaskevi commited on
Commit
56397d8
1 Parent(s): b4c5c5b

Update chatbot.py

Browse files
Files changed (1) hide show
  1. chatbot.py +2 -2
chatbot.py CHANGED
@@ -67,13 +67,13 @@ def demo_miny_memory():
67
  memory = ConversationBufferMemory(memory_key="history", llm = llm)
68
  return memory
69
 
70
- def demo_chain(input_text, memory):
71
  PROMPT = ChatPromptTemplate.from_template(template)
72
  conversation = ConversationChain(
73
  prompt=PROMPT,
74
  llm=llm,
75
  #verbose=langchain.globals.get_verbose(),
76
- memory=memory
77
  )
78
 
79
  chat_reply = conversation.invoke(input=input_text, max_new_tokens = 100)
 
67
  memory = ConversationBufferMemory(memory_key="history", llm = llm)
68
  return memory
69
 
70
+ def demo_chain(input_text):
71
  PROMPT = ChatPromptTemplate.from_template(template)
72
  conversation = ConversationChain(
73
  prompt=PROMPT,
74
  llm=llm,
75
  #verbose=langchain.globals.get_verbose(),
76
+ memory=demo_miny_memory()
77
  )
78
 
79
  chat_reply = conversation.invoke(input=input_text, max_new_tokens = 100)