toll5

app.py

@@ -641,23 +641,23 @@ demo: Optional[gr.Interface] = None  # Assign your gradio demo to this variable
 return_type = List[Hit]
 
 ## YOUR_CODE_STARTS_HERE
-csc_bm25_index = CSCBM25Index.build_from_documents(
+bm25_index = BM25Index.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")
+bm25_index.save("output/bm25_index")
 
 def search(query: str) -> List[Hit]:
-    csc_bm25_index = CSCBM25Retriever(index_dir="output/csc_bm25_index")
-    result = csc_bm25_index.retrieve(query)
+    bm25_index = CSCBM25Retriever(index_dir="output/bm25_index")
+    result = bm25_index.retrieve(query)
 
     l : return_type = []
     for cid, score in result.items():
-      docid = csc_bm25_index.index.cid2docid[cid]
-      text = csc_bm25_index.index.doc_texts[docid]
+      docid = bm25_index.index.cid2docid[cid]
+      text = bm25_index.index.doc_texts[docid]
 
       l.append(Hit(cid=cid, score=score, text=text))