some changes

app.py

@@ -11,6 +11,7 @@ from langchain.text_splitter import RecursiveCharacterTextSplitter
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 
+# Updated Prompt Template
 persist_directory = 'db'
 embeddings_model_name = 'sentence-transformers/all-MiniLM-L6-v2'
 
@@ -34,6 +35,8 @@ def get_vector_store(target_source_chunks):
     return retriver
 
 def get_conversation_chain(retriever):
+    tokenizer = AutoTokenizer.from_pretrained("red1xe/Llama-2-7B-codeGPT")
+    model = AutoModelForCausalLM.from_pretrained("red1xe/Llama-2-7B-codeGPT")
     memory = ConversationBufferMemory(memory_key='chat_history', return_messages=True,)
     chain = RetrievalQA.from_llm(
         llm=model,