File size: 301 Bytes
a5d886c 12003d8 | 1 2 3 4 5 6 7 8 9 | from sentence_transformers import SentenceTransformer
class EmbeddingService:
def __init__(self):
self.model = SentenceTransformer("all-MiniLM-L6-v2")
def embed(self, texts: list[str]) -> list[list[float]]:
return self.model.encode(texts, normalize_embeddings=True).tolist()
|