File size: 268 Bytes
c1e438c
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
from sentence_transformers import SentenceTransformer

class Embedder:
    def __init__(self, model="sentence-transformers/all-MiniLM-L6-v2"):
        self.model = SentenceTransformer(model)

    def encode(self, text):
        return self.model.encode(text).tolist()