sentence-transformers How to use sobamchan/roberta-base-mean-softmax-250 with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("sobamchan/roberta-base-mean-softmax-250")
sentences = [
"Two women having drinks and smoking cigarettes at the bar.",
"Women are celebrating at a bar.",
"Two kids are outdoors.",
"The four girls are attending the street festival."
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4] 
