sentence-transformers How to use djsull/sentence-roberta-small with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("djsull/sentence-roberta-small")
sentences = [
"백마를 이끄는 여자",
"갈색 말을 타고 있는 여자",
"남자와 여자가 키스하고 있다.",
"남자가 칼로 물병을 썰고 있다"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4] 
