How to use sobamchan/bert-base-uncased-mean-350 with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("sobamchan/bert-base-uncased-mean-350") sentences = [ "A man is jumping unto his filthy bed.", "A young male is looking at a newspaper while 2 females walks past him.", "The bed is dirty.", "The man is on the moon." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4]
