sentence-transformers How to use richinfoai/ritrieve_zh_v1 with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("richinfoai/ritrieve_zh_v1")
sentences = [
"那是 個快樂的人",
"那是 條快樂的狗",
"那是 個非常幸福的人",
"今天是晴天"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4] 
