makcrx commited on
Commit
806fb28
·
1 Parent(s): b3d0802

increase bi-encoder samples

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ db = FAISS.load_local('faiss_qa_2023-08-20', embeddings)
11
  def main(query):
12
  query = query.lower()
13
  query_keywords, query = extract_keywords2(query)
14
- result_docs = db.similarity_search_with_score(query, k=20)
15
  print(query, query_keywords)
16
 
17
  if len(query_keywords) > 0:
 
11
  def main(query):
12
  query = query.lower()
13
  query_keywords, query = extract_keywords2(query)
14
+ result_docs = db.similarity_search_with_score(query, k=50)
15
  print(query, query_keywords)
16
 
17
  if len(query_keywords) > 0: