BanUrsus commited on
Commit
b043f69
·
verified ·
1 Parent(s): 2cd74f5

Update app.py

Browse files

change load pkl

Files changed (1) hide show
  1. app.py +6 -0
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)