How to use sobamchan/roberta-base-mean-150 with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("sobamchan/roberta-base-mean-150") 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]
What is a pickle import?
How to fix it?