Update app.py

app.py

@@ -2,7 +2,8 @@ import gradio as gr
 import spaces
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from langchain.llms.base import LLM
-from langchain.chains import LLMChain
+from langchain.memory import ConversationBufferMemory
+from langchain.chains import LLMChain, ConversationChain
 from langchain.prompts import PromptTemplate, ChatPromptTemplate
 
 @spaces.GPU