Update app.py

app.py

@@ -532,25 +532,18 @@ class Hit(TypedDict):
   score: float
   text: str
 
-# Best b 0.9
-# Best k1: 0.4
-best_b = 0.9
-best_k1 = 0.4 
-sciq = load_sciq()
-csc_bm25_index = CSCBM25Index.build_from_documents(
-    documents=iter(sciq.corpus),
-    ndocs=12160,
-    show_progress_bar=True,
-    k1=best_k1,
-    b=best_b
-)
-csc_bm25_index.save("output/csc_bm25_index")
-
 demo: Optional[gr.Interface] = None  # Assign your gradio demo to this variable
 return_type = List[Hit]
 
 ## YOUR_CODE_STARTS_HERE
-csc_bm25_retriever = CSCBM25Retriever(index_dir="output/csc_bm25_index")
+# Use default b, k1
+csc_bm25_index = CSCBM25Index.build_from_documents(
+    documents=iter(sciq.corpus),
+    ndocs=12160,
+    show_progress_bar=True
+  )
+csc_bm25_index.save("output/csc_bm25_index_default")
+csc_bm25_retriever = CSCBM25Retriever(index_dir="output/csc_bm25_index_default")
 doc2text = {doc.collection_id: doc.text for doc in sciq.corpus}
 
 def retrieve(query: str) -> List[Hit]: