deneme
Browse files
app.py
CHANGED
|
@@ -30,10 +30,9 @@ if files:
|
|
| 30 |
embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
|
| 31 |
vectorstore = FAISS.from_texts(chunks, embeddings)
|
| 32 |
memory = ConversationBufferMemory(memory_key='chat_history', return_messages=True,)
|
| 33 |
-
llm = AutoModelForCausalLM.from_pretrained("
|
| 34 |
-
chain = RetrievalQA.
|
| 35 |
llm=llm,
|
| 36 |
-
chain_type="retrieval-qa",
|
| 37 |
retriever=vectorstore.as_retriever(),
|
| 38 |
memory=memory,
|
| 39 |
)
|
|
|
|
| 30 |
embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
|
| 31 |
vectorstore = FAISS.from_texts(chunks, embeddings)
|
| 32 |
memory = ConversationBufferMemory(memory_key='chat_history', return_messages=True,)
|
| 33 |
+
llm = AutoModelForCausalLM.from_pretrained("red1xe/Llama-2-7B-codeGPT")
|
| 34 |
+
chain = RetrievalQA.from_llm(
|
| 35 |
llm=llm,
|
|
|
|
| 36 |
retriever=vectorstore.as_retriever(),
|
| 37 |
memory=memory,
|
| 38 |
)
|