Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -58,7 +58,7 @@ corpus_embeddings.shape
|
|
58 |
def find_sentences(query):
|
59 |
query_embedding = model.encode(query)
|
60 |
hits = util.semantic_search(query_embedding, corpus_embeddings, top_k=5)
|
61 |
-
|
62 |
|
63 |
|
64 |
corpus_id = hit['corpus_id']
|
|
|
58 |
def find_sentences(query):
|
59 |
query_embedding = model.encode(query)
|
60 |
hits = util.semantic_search(query_embedding, corpus_embeddings, top_k=5)
|
61 |
+
hit = hits[0][0]
|
62 |
|
63 |
|
64 |
corpus_id = hit['corpus_id']
|