sentence-transformers How to use johnnyboycurtis/ModernBERT-small-sts with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("johnnyboycurtis/ModernBERT-small-sts")
sentences = [
"A modern bathtub in a bathroom is displayed.",
"Different types of tiles are on the walls, floor and tub.",
"A man sitting on a park bench looking towards a fountain and sculpture.",
"A bathroom with a shower and his and her sinks."
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4] 
