File size: 252 Bytes
c68bbe9 | 1 2 3 4 5 6 7 8 9 | from sentence_transformers import SentenceTransformer
class EmbeddingTool:
def __init__(self):
self.model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
def embed(self, texts):
return self.model.encode(texts) |