red1xe commited on
Commit
314ff94
1 Parent(s): a39564e
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ if files:
31
  db = FAISS.from_texts(chunks, embeddings)
32
  memory = ConversationBufferMemory(memory_key='chat_history', return_messages=True)
33
  chain = RetrievalQA.from_llm(
34
- llm=AutoModelForCausalLM.from_pretrained("red1xe/falcon-7b-codeGPT-3K"),
35
  memory=memory,
36
  retriever=db.as_retriever(search_kwargs={"k": 3}),
37
  )
 
31
  db = FAISS.from_texts(chunks, embeddings)
32
  memory = ConversationBufferMemory(memory_key='chat_history', return_messages=True)
33
  chain = RetrievalQA.from_llm(
34
+ llm=AutoModelForCausalLM.from_pretrained("red1xe/Llama-2-7B-codeGPT"),
35
  memory=memory,
36
  retriever=db.as_retriever(search_kwargs={"k": 3}),
37
  )