Spaces:
Sleeping
Sleeping
cfgd
Browse files
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.
|
| 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]
|