sentence-transformers How to use haf1g/result_model with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("haf1g/result_model")
sentences = [
"A man, woman, and child enjoying themselves on a beach.",
"A family of three is at the beach.",
"There are two woman in this picture.",
"There are children present"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4]