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']
|