sentence-transformers How to use sentence-transformers-testing/all-nli-bert-tiny-dense with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("sentence-transformers-testing/all-nli-bert-tiny-dense")
sentences = [
"Tall man being stopped by an officer.",
"The man is short.",
"There is a tall man.",
"Male in brown leather jacket and tight black slacks, looking down at his phone"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4] 
