Update chatbot.py
Browse files- 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.
|
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]
|