Spaces:

VcRlAgent
/

workwise-backend-gpu

Sleeping

VcRlAgent commited on Nov 10, 2025

Commit

d56b9a1

1 Parent(s): 3eb548d

New Hardware

Files changed (1) hide show

app/services/reranker.py CHANGED Viewed

@@ -9,7 +9,7 @@ class RerankerService:
     Cross-Encoder based re-ranker for improving top-k retrieval precision.
     """
     def __init__(self, model_name: str = "cross-encoder/ms-marco-MiniLM-L-6-v2"):
-        logger.info(f"Loading reranker model: {model_name}")
         self.model = CrossEncoder(model_name)
     def rerank(self, query: str, results: list, top_k: int = 5) -> list:

     Cross-Encoder based re-ranker for improving top-k retrieval precision.
     """
     def __init__(self, model_name: str = "cross-encoder/ms-marco-MiniLM-L-6-v2"):
+        logger.info(f"[RERANKER] Loading reranker model: {model_name}")
         self.model = CrossEncoder(model_name)
     def rerank(self, query: str, results: list, top_k: int = 5) -> list: