Build BM25 index from documents and save to output directory

app.py

@@ -363,17 +363,16 @@ class Hit(TypedDict):
 
 demo: Optional[gr.Interface] = None  # Assign your gradio demo to this variable
 return_type = List[Hit]
+
+## YOUR_CODE_STARTS_HERE
 bm25_index = BM25Index.build_from_documents(
     documents=iter(sciq.corpus),
     ndocs=12160,
     show_progress_bar=True,
-    k1=0.9,
-    b=0.4,
 )
 bm25_index.save("output/bm25_index")
 bm25_retriever = BM25Retriever(index_dir="output/bm25_index")
 
-## YOUR_CODE_STARTS_HERE
 def retrieve(query: str, topk: int = 10) -> return_type:
     ranking = bm25_retriever.retrieve(query=query, topk=topk)
     hits = []