KvrParaskevi
commited on
Update chatbot.py
Browse files- 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
|
71 |
PROMPT = ChatPromptTemplate.from_template(template)
|
72 |
conversation = ConversationChain(
|
73 |
prompt=PROMPT,
|
74 |
llm=llm,
|
75 |
#verbose=langchain.globals.get_verbose(),
|
76 |
-
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)
|