Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 from langchain.llms import LlamaCpp
 from langchain.memory import ConversationBufferMemory
-from langchain.chains import ConversationalChain
+from langchain.chains import ConversationalRetrievalChain
 
 # Streamlit page configuration
 st.set_page_config(page_title="Simple AI Chatbot")
@@ -10,8 +10,8 @@ st.header("Simple AI Chatbot")
 # Initialize the Language Model Chain
 @st.experimental_singleton
 def initialize_chain():
-    n_gpu_layers = 40
-    n_batch = 2048
+    n_gpu_layers = 20
+    n_batch = 1024
 
     llm = LlamaCpp(
         model_path="models/mistral-7b-instruct-v0.1.Q5_0.gguf",