Namitg02 commited on
Commit
f51e056
·
verified ·
1 Parent(s): 3c80115

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -56,10 +56,10 @@ retriever = vectordb.as_retriever(
56
  search_type="similarity", search_kwargs={"k": 2}
57
  )
58
 
59
- #from langchain.chains import RetrievalQA
60
  from langchain_core.prompts import ChatPromptTemplate
61
 
62
- from langchain.chains import create_retrieval_chain
63
  from langchain.chains.combine_documents import create_stuff_documents_chain
64
  #from langchain import hub
65
 
@@ -68,7 +68,7 @@ READER_MODEL="HuggingFaceH4/zephyr-7b-beta"
68
  #HuggingFaceH4/zephyr-7b-beta
69
  #READER_MODEL=Ollama(model="meta-llama/Meta-Llama-Guard-2-8B")
70
  #qa = ConversationalRetrievalChain.from_llm(llm=READER_MODEL,retriever=retriever,memory=memory)
71
- #qa = RetrievalQA.from_chain_type(llm=READER_MODEL,chain_type="map_reduce",retriever=retriever,verbose=True)
72
 
73
  #retrieval_qa_chat_prompt = hub.pull("langchain-ai/retrieval-qa-chat")
74
 
 
56
  search_type="similarity", search_kwargs={"k": 2}
57
  )
58
 
59
+ from langchain.chains import RetrievalQA
60
  from langchain_core.prompts import ChatPromptTemplate
61
 
62
+ #from langchain.chains import create_retrieval_chain
63
  from langchain.chains.combine_documents import create_stuff_documents_chain
64
  #from langchain import hub
65
 
 
68
  #HuggingFaceH4/zephyr-7b-beta
69
  #READER_MODEL=Ollama(model="meta-llama/Meta-Llama-Guard-2-8B")
70
  #qa = ConversationalRetrievalChain.from_llm(llm=READER_MODEL,retriever=retriever,memory=memory)
71
+ qa = RetrievalQA.from_chain_type(llm=READER_MODEL,chain_type="map_reduce",retriever=retriever,verbose=True)
72
 
73
  #retrieval_qa_chat_prompt = hub.pull("langchain-ai/retrieval-qa-chat")
74