venthan commited on
Commit
4decf49
·
1 Parent(s): fcaa09e

change k results to 5

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ class AASLDConversationalAgent():
38
  )
39
  self.doc_search = FAISS.from_documents(pages, embeddings_model)
40
  self.doc_retriever = self.doc_search.as_retriever(
41
- search_type="mmr", search_kwargs={"k": 10})
42
  self.llm = ChatOpenAI(temperature = 0, model_name=LLM_MODEL)
43
  tool = create_retriever_tool(
44
  self.doc_retriever,
 
38
  )
39
  self.doc_search = FAISS.from_documents(pages, embeddings_model)
40
  self.doc_retriever = self.doc_search.as_retriever(
41
+ search_type="mmr", search_kwargs={"k": 5})
42
  self.llm = ChatOpenAI(temperature = 0, model_name=LLM_MODEL)
43
  tool = create_retriever_tool(
44
  self.doc_retriever,