How to use tomaarsen/distilroberta-base-nli-angle with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("tomaarsen/distilroberta-base-nli-angle") sentences = [ "A man dressed in yellow rescue gear walks in a field.", "A person messes with some papers.", "The man is outdoors.", "The man is bowling." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4]
