Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -46,12 +46,10 @@ for sent in context.split("."):
|
|
| 46 |
sentence_count.append(len(sentences))
|
| 47 |
for _,s in enumerate(sentences):
|
| 48 |
corpus.append(s)
|
| 49 |
-
|
| 50 |
-
|
| 51 |
-
|
| 52 |
-
corpus_embeddings = np.load('task_embeddings_updated_msmarco-distilbert-base-v4.npy')
|
| 53 |
-
|
| 54 |
|
|
|
|
|
|
|
|
|
|
| 55 |
|
| 56 |
def find_sentences(query):
|
| 57 |
query_embedding = model.encode(query)
|
|
|
|
| 46 |
sentence_count.append(len(sentences))
|
| 47 |
for _,s in enumerate(sentences):
|
| 48 |
corpus.append(s)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 49 |
|
| 50 |
+
corpus_embeddings = model.encode(corpus)
|
| 51 |
+
np.save('saved_embeddings.npy', corpus_embeddings)
|
| 52 |
+
corpus_embeddings = np.load('saved_embeddings.npy')
|
| 53 |
|
| 54 |
def find_sentences(query):
|
| 55 |
query_embedding = model.encode(query)
|