Update app.py
Browse files
app.py
CHANGED
@@ -64,9 +64,9 @@ from langchain.chains.combine_documents import create_stuff_documents_chain
|
|
64 |
from langchain.chains import create_retrieval_chain
|
65 |
|
66 |
|
67 |
-
READER_MODEL="HuggingFaceH4/zephyr-7b-beta"
|
68 |
#HuggingFaceH4/zephyr-7b-beta
|
69 |
-
|
70 |
|
71 |
#qa = ConversationalRetrievalChain.from_llm(llm=READER_MODEL,retriever=retriever,memory=memory)
|
72 |
#qa = RetrievalQA.from_chain_type(llm=READER_MODEL,retriever=retriever)
|
|
|
64 |
from langchain.chains import create_retrieval_chain
|
65 |
|
66 |
|
67 |
+
#READER_MODEL="HuggingFaceH4/zephyr-7b-beta"
|
68 |
#HuggingFaceH4/zephyr-7b-beta
|
69 |
+
READER_MODEL=Ollama(model="meta-llama/Meta-Llama-Guard-2-8B")
|
70 |
|
71 |
#qa = ConversationalRetrievalChain.from_llm(llm=READER_MODEL,retriever=retriever,memory=memory)
|
72 |
#qa = RetrievalQA.from_chain_type(llm=READER_MODEL,retriever=retriever)
|