Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -50,8 +50,6 @@ for sent in context.split("."):
|
|
50 |
|
51 |
|
52 |
corpus_embeddings = np.load('task_embeddings_msmarco-distilbert-base-v4.npy')
|
53 |
-
corpus_embeddings.shape
|
54 |
-
|
55 |
|
56 |
|
57 |
|
@@ -59,8 +57,10 @@ def find_sentences(query):
|
|
59 |
query_embedding = model.encode(query)
|
60 |
hits = util.semantic_search(query_embedding, corpus_embeddings, top_k=1)
|
61 |
hit = hits[0][0]
|
|
|
62 |
corpus_id = hit['corpus_id']
|
63 |
saved = corpus[corpus_id]
|
|
|
64 |
# Find source document based on sentence index
|
65 |
count = 0
|
66 |
for idx, c in enumerate(sentence_count):
|
|
|
50 |
|
51 |
|
52 |
corpus_embeddings = np.load('task_embeddings_msmarco-distilbert-base-v4.npy')
|
|
|
|
|
53 |
|
54 |
|
55 |
|
|
|
57 |
query_embedding = model.encode(query)
|
58 |
hits = util.semantic_search(query_embedding, corpus_embeddings, top_k=1)
|
59 |
hit = hits[0][0]
|
60 |
+
print(hit)
|
61 |
corpus_id = hit['corpus_id']
|
62 |
saved = corpus[corpus_id]
|
63 |
+
print(saved)
|
64 |
# Find source document based on sentence index
|
65 |
count = 0
|
66 |
for idx, c in enumerate(sentence_count):
|