hellopahe commited on
Commit
c215129
1 Parent(s): 4d46ad9

return 10 results for simi-check

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ def similarity_search(queries, doc):
49
  query_list = queries.split('\n')
50
 
51
  corpus_embeddings = embedder.encode(doc_list, convert_to_tensor=True)
52
- top_k = min(5, len(doc_list))
53
  output = ""
54
  for query in query_list:
55
  query_embedding = embedder.encode(query, convert_to_tensor=True)
 
49
  query_list = queries.split('\n')
50
 
51
  corpus_embeddings = embedder.encode(doc_list, convert_to_tensor=True)
52
+ top_k = min(10, len(doc_list))
53
  output = ""
54
  for query in query_list:
55
  query_embedding = embedder.encode(query, convert_to_tensor=True)