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]
