NimrodDev commited on
Commit
3dd74e6
·
1 Parent(s): 54f0cf8
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -80,7 +80,7 @@ def atomic_retriever():
80
  bm25 = BM25Okapi(tokenized)
81
 
82
  def search(q: str) -> List[Document]:
83
- dense_hits = dense.get_relevant_documents(q)
84
  scores = bm25.get_scores(re.findall(r"\w+", q.lower()))
85
  top = np.argsort(scores)[-5:][::-1]
86
  bm25_hits = [docs[i] for i in top if scores[i] > 0]
 
80
  bm25 = BM25Okapi(tokenized)
81
 
82
  def search(q: str) -> List[Document]:
83
+ dense_hits = dense.invoke(q) # v1 API
84
  scores = bm25.get_scores(re.findall(r"\w+", q.lower()))
85
  top = np.argsort(scores)[-5:][::-1]
86
  bm25_hits = [docs[i] for i in top if scores[i] > 0]