Spaces:

merve
/

taskmaster

Sleeping

merve HF Staff commited on Jan 25, 2022

Commit

2c706e6

1 Parent(s): 04b2fae

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -47,7 +47,7 @@ for sent in context.split("."):
     for _,s in enumerate(sentences):
         corpus.append(s)
-corpus_embeddings = np.load('task_embeddings_msmarco-distilbert-base-v4.npy')
 def find_sentences(query):
     query_embedding = model.encode(query)

     for _,s in enumerate(sentences):
         corpus.append(s)
+corpus_embeddings = model.encode(corpus)
 def find_sentences(query):
     query_embedding = model.encode(query)