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)
|