merve HF Staff commited on
Commit
04b2fae
·
1 Parent(s): 43d8f7d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -47,9 +47,7 @@ for sent in context.split("."):
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)
 
47
  for _,s in enumerate(sentences):
48
  corpus.append(s)
49
 
50
+ corpus_embeddings = np.load('task_embeddings_msmarco-distilbert-base-v4.npy')
 
 
51
 
52
  def find_sentences(query):
53
  query_embedding = model.encode(query)