sentence-transformers How to use omkar334/bert-base-uncased-retromae with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("omkar334/bert-base-uncased-retromae")
sentences = [
"Many people walk away from the camera down a cobblestone alley.",
"A man types away in his office.",
"the boy is ten years old",
"Two kids are laughing in the grass."
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4] 
