Update app.py
Browse fileschange load pkl
app.py
CHANGED
@@ -345,6 +345,12 @@ return_type = List[Hit]
|
|
345 |
|
346 |
def search(query: str) -> List[Hit]:
|
347 |
|
|
|
|
|
|
|
|
|
|
|
|
|
348 |
csc_bm25_retriever = CSCBM25Retriever(index_dir="output/csc_bm25_index")
|
349 |
|
350 |
top_docs = csc_bm25_retriever.retrieve(query)
|
|
|
345 |
|
346 |
def search(query: str) -> List[Hit]:
|
347 |
|
348 |
+
csc_bm25_index = CSCBM25Index.build_from_documents(
|
349 |
+
documents=iter(sciq.corpus),
|
350 |
+
ndocs=12160,
|
351 |
+
show_progress_bar=True,
|
352 |
+
)
|
353 |
+
csc_bm25_index.save("output/csc_bm25_index")
|
354 |
csc_bm25_retriever = CSCBM25Retriever(index_dir="output/csc_bm25_index")
|
355 |
|
356 |
top_docs = csc_bm25_retriever.retrieve(query)
|