def __call__()

in src/huggingface_inference_toolkit/sentence_transformers_utils.py [0:0]

• 5 lines of code
• 1 McCabe index (conditional complexity)

    def __call__(self, source_sentence: str, sentences: List[str]) -> Dict[str, float]:
        embeddings1 = self.model.encode(source_sentence, convert_to_tensor=True)
        embeddings2 = self.model.encode(sentences, convert_to_tensor=True)
        similarities = util.pytorch_cos_sim(embeddings1, embeddings2).tolist()[0]
        return {"similarities": similarities}