How to use from the
Use from the
sentence-transformers library
from sentence_transformers import SentenceTransformer

model = SentenceTransformer("NeuML/bert-tiny-prompts")

sentences = [
    "That is a happy person",
    "That is a happy dog",
    "That is a very happy person",
    "Today is a sunny day"
]
embeddings = model.encode(sentences)

similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4]

BERT Tiny Prompts

This model is a copy of stsb-bert-tiny-safetensors with prompts added for testing. It's intended for use with unit tests.

Downloads last month
964
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support