namngo commited on
Commit
10cc155
·
verified ·
1 Parent(s): e5c437d

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -42,7 +42,7 @@ def generate_answer(model, tokenizer, context, question, max_length=256):
42
 
43
  # print(model)
44
 
45
- def find_context(pos_sentences,question,model,embedings='similarity_embeddings.npy'):
46
  data = np.load(embedings)
47
  pos_embeddings = data["embeddings"]
48
  query_embedding = model.encode(question)
 
42
 
43
  # print(model)
44
 
45
+ def find_context(pos_sentences,question,model,embedings='similarity_embeddings.npz'):
46
  data = np.load(embedings)
47
  pos_embeddings = data["embeddings"]
48
  query_embedding = model.encode(question)