Spaces:
Build error
Build error
hellopahe
commited on
Commit
•
c215129
1
Parent(s):
4d46ad9
return 10 results for simi-check
Browse files
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(
|
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)
|