File size: 169 Bytes
bb4e28f
 
 
 
 
 
1
2
3
4
5
6
7
from sentence_transformers import SentenceTransformer

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

def get_embedding(text: str):
    return model.encode([text])[0]