sentence-transformers How to use tomaarsen/mpnet-base-allnli with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("tomaarsen/mpnet-base-allnli")
sentences = [
"A taxi SUV drives past an urban construction site, as a man walks down the street in the other direction.",
"The woman is walking down the street with high heels.",
"A man is reading documents in a binder.",
"A man is chasing an SUV that is going in the same direction as him."
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4] 
