Update app.py

app.py

@@ -48,6 +48,11 @@ from langchain import PromptTemplate
 from langchain import LLMChain
 from langchain.llms import CTransformers
 
+from langchain.memory import ConversationBufferWindowMemory
+
+memory = ConversationBufferWindowMemory()
+
+
 # Update the model path to use the Hugging Face model identifier
 llm = CTransformers(model='TheBloke/Llama-2-7B-Chat-GGML',
                     model_file='llama-2-7b-chat.ggmlv3.q3_K_S.bin',
@@ -56,6 +61,12 @@ llm = CTransformers(model='TheBloke/Llama-2-7B-Chat-GGML',
                             'temperature': 0.01}
                     )
 
+# convo = ConversationChain(
+#     llm=llm,
+#     memory=memory
+# )
+# convo.run("Who won the first cricket world cup?")
+
 def greet(my_prompt):
 
     B_INST, E_INST = "[INST]", "[/INST]"
@@ -90,7 +101,7 @@ def greet(my_prompt):
     prompt = PromptTemplate(template=template, input_variables=["text"])
     
     
-    LLM_Chain = LLMChain(prompt=prompt, llm=llm)
+    LLM_Chain = LLMChain(prompt=prompt, llm=llm, memory=memory)
     
     # print(LLM_Chain.run("How are you"))