Update app.py

app.py

@@ -6,7 +6,7 @@ from sentence_transformers import SentenceTransformer
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 from beir import util
 from beir.datasets.data_loader import GenericDataLoader
-from beir import EvaluateRetrieval
+#from beir import EvaluateRetrieval
 
 
 # Function to load the dataset
@@ -53,6 +53,7 @@ def rerank_passages(retrieved_indices, corpus, queries):
     return reranked_passages
 
 # Function for evaluation
+""""
 def evaluate(qrels, retrieved_indices, reranked_passages, queries):
     evaluator = EvaluateRetrieval()
     
@@ -73,7 +74,7 @@ def evaluate(qrels, retrieved_indices, reranked_passages, queries):
     
     ndcg_score_stage2 = evaluator.evaluate(qrels, results_stage2, [10])['NDCG@10']
     return ndcg_score_stage1, ndcg_score_stage2
-
+    """
 # Streamlit app
 def main():
     st.title("Multi-Stage Text Retrieval Pipeline")
@@ -92,10 +93,11 @@ def main():
         st.success("Reranking completed!")
         st.write("Reranked passages:", reranked_passages)
 
-    if st.button("Evaluate"):
+    """if st.button("Evaluate"):
         ndcg_score_stage1, ndcg_score_stage2 = evaluate(qrels, retrieved_indices, reranked_passages, queries)
         st.write(f"NDCG@10 for Stage 1 (Candidate Retrieval): {ndcg_score_stage1}")
         st.write(f"NDCG@10 for Stage 2 (Reranking): {ndcg_score_stage2}")
+        """
 
 if __name__ == "__main__":
     main()