Amelia-James commited on
Commit
b52f1b7
·
verified ·
1 Parent(s): a76753a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,7 +61,7 @@ Helpful Answer (based on the documents, 100 words max):"""
61
  QA_CHAIN_PROMPT = PromptTemplate.from_template(template)
62
 
63
  # Improved retriever settings
64
- retriever = vectordb.as_retriever(search_type="mmr", search_kwargs={"k": 5}) # Maximal Marginal Relevance for better diversity
65
 
66
  # Build the QA chain with improved relevance
67
  qa_chain = RetrievalQA.from_chain_type(
 
61
  QA_CHAIN_PROMPT = PromptTemplate.from_template(template)
62
 
63
  # Improved retriever settings
64
+ retriever = vectordb.as_retriever(search_type="mmr", search_kwargs={"k": 1}) # Maximal Marginal Relevance for better diversity
65
 
66
  # Build the QA chain with improved relevance
67
  qa_chain = RetrievalQA.from_chain_type(