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]
The community tab is the place to discuss and collaborate with the HF community!
