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]
