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