sentence-transformers How to use kasraarabi/finetuned-caption-embedding with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("kasraarabi/finetuned-caption-embedding")
sentences = [
"an elephant with a leaf on its back",
"an elephant is walking through the woods",
"a white truck with a white sign on it",
"a bathroom with a tub and sink"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4] 
