How to use trbeers/distilroberta-base-nli-v2 with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("trbeers/distilroberta-base-nli-v2") sentences = [ "An Indian woman is washing and cleaning dirty laundry at a lake and in the background is a kid who appears to have jumped into the lake.", "An Indian woman is doing her laundry in a lake.", "An Indian woman is putting her laundry into the machine.", "A girl is playing with a Slinky." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4]
