show progress bar

Author: TideDra

src/zotero_arxiv_daily/reranker/local.py

@@ -24,7 +24,7 @@ def get_similarity_score(self, s1: list[str], s2: list[str]) -> np.ndarray:
             encode_kwargs = self.config.reranker.local.encode_kwargs
         else:
             encode_kwargs = {}
-        s1_feature = encoder.encode(s1,**encode_kwargs)
-        s2_feature = encoder.encode(s2,**encode_kwargs)
+        s1_feature = encoder.encode(s1,**encode_kwargs,show_progress_bar=True)
+        s2_feature = encoder.encode(s2,**encode_kwargs,show_progress_bar=True)
         sim = encoder.similarity(s1_feature, s2_feature)
         return sim.numpy()