justest's picture
Duplicate from justest/embeddings-api
b19a023
raw history blame
No virus
292 Bytes
from sentence_transformers import SentenceTransformer
model = SentenceTransformer('moka-ai/m3e-base')
# model = SentenceTransformer('nghuyong/ernie-3.0-base-zh')
def encode(text: str):
text = text.replace("\n", " ")
return model.encode([text], normalize_embeddings=True).tolist()[0]