1

app.py

@@ -30,7 +30,7 @@ if files:
         embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
         vectorstore = FAISS.from_texts(chunks, embeddings)
         memory = ConversationBufferMemory(memory_key='chat_history', return_messages=True,)
-        llm = AutoModelForCausalLM.from_pretrained("red1xe/Llama-2-7B-codeGPT")
+        llm = AutoModelForCausalLM.from_pretrained("TinyPixel/Llama-2-7B-bf16-sharded")
         chain = RetrievalQA.from_chain_type(
             llm=llm,
             chain_type="retrieval-qa",