KvrParaskevi
commited on
Update chatbot.py
Browse files- 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
|
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,
|