merve HF Staff commited on
Commit
a24cd69
·
1 Parent(s): 7cbadd4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
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)