Upload app_federal.py
Browse files- app_federal.py +1 -1
app_federal.py
CHANGED
|
@@ -228,7 +228,7 @@ def load_bm25():
|
|
| 228 |
return retriever, bm25_tokenizer
|
| 229 |
|
| 230 |
def run_bm25(query):
|
| 231 |
-
query_tokens = bm25_tokenizer.tokenize(query
|
| 232 |
results, scores = retriever.retrieve(query_tokens, k=5)
|
| 233 |
return results[0]
|
| 234 |
|
|
|
|
| 228 |
return retriever, bm25_tokenizer
|
| 229 |
|
| 230 |
def run_bm25(query):
|
| 231 |
+
query_tokens = bm25_tokenizer.tokenize(query)
|
| 232 |
results, scores = retriever.retrieve(query_tokens, k=5)
|
| 233 |
return results[0]
|
| 234 |
|