KvrParaskevi commited on
Commit
f8d16dc
1 Parent(s): d3ba2cf

Update chatbot.py

Browse files
Files changed (1) hide show
  1. chatbot.py +5 -4
chatbot.py CHANGED
@@ -62,12 +62,13 @@ def load_pipeline():
62
  llm = load_pipeline()
63
 
64
  def demo_miny_memory():
65
- prompt = ChatPromptTemplate.from_template(template)
66
- memory = ConversationBufferMemory(memory_key="history", llm = llm, prompt = prompt)
67
  return memory
68
 
69
- def demo_chain(input_text):
70
- PROMPT = ChatPromptTemplate.from_template(template)
 
71
  conversation = ConversationChain(
72
  prompt=PROMPT,
73
  llm=llm,
 
62
  llm = load_pipeline()
63
 
64
  def demo_miny_memory():
65
+ #prompt = ChatPromptTemplate.from_template(template)
66
+ memory = ConversationBufferMemory(memory_key="history", llm = llm)
67
  return memory
68
 
69
+ def demo_chain(input_text,history):
70
+ #PROMPT = ChatPromptTemplate.from_template(template)
71
+ PROMPT = ChatPromptTemplate.from_messages(history)
72
  conversation = ConversationChain(
73
  prompt=PROMPT,
74
  llm=llm,