File size: 199 Bytes
345000b
 
 
 
 
 
1
2
3
4
5
6
7
from sentence_transformers import SentenceTransformer

model = SentenceTransformer("all-MiniLM-L6-v2")

def embed_texts(texts: list[str]):
    return model.encode(texts, convert_to_numpy=True)