Update app.py

app.py

@@ -44,7 +44,8 @@ Human: {input}
 AI:"""
 prompt = PromptTemplate(template=template, input_variables=["history", "input"])
 
-llm_chain = LLMChain(prompt=prompt, llm=llm)
+memory = ConversationBufferMemory(memory_key="history", llm = llm, prompt = prompt)
+llm_chain = ConversationChain(prompt=prompt, llm=llm, memory = memory)
 
 with gr.Blocks() as demo:
     chatbot = gr.Chatbot()
@@ -57,7 +58,7 @@ with gr.Blocks() as demo:
     @spaces.GPU
     def bot(history):
         print("Question: ", history[-1][0])
-        bot_message = llm_chain.run(question=history[-1][0])
+        bot_message = llm_chain.invoke(question=history[-1][0])
         print("Response: ", bot_message)
         history[-1][1] = ""
         history[-1][1] += bot_message