sofarikasid commited on
Commit
4a2b019
1 Parent(s): 1e64423

Synced repo using 'sync_with_huggingface' Github Action

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -52,7 +52,7 @@ def similar_offer(llm, prompt, db):
52
  qa_chain = RetrievalQA.from_chain_type(
53
  llm=llm,
54
  chain_type="stuff",
55
- retriever=db.as_retriever(search_kwargs={"k": 20}),
56
  return_source_documents=True,
57
  chain_type_kwargs={"prompt": prompt},
58
  )
@@ -89,4 +89,4 @@ def main():
89
 
90
 
91
  if __name__ == "__main__":
92
- main()
 
52
  qa_chain = RetrievalQA.from_chain_type(
53
  llm=llm,
54
  chain_type="stuff",
55
+ retriever=db.as_retriever(search_kwargs={"k": 5}),
56
  return_source_documents=True,
57
  chain_type_kwargs={"prompt": prompt},
58
  )
 
89
 
90
 
91
  if __name__ == "__main__":
92
+ main()