sentence-transformers How to use rufimelo/Legal-BERTimbau-sts-base with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("rufimelo/Legal-BERTimbau-sts-base")
sentences = [
"O advogado apresentou as provas ao juíz.",
"O juíz leu as provas.",
"O juíz leu o recurso.",
"O juíz atirou uma pedra."
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4]
