How to use swardiantara/bert-tiny-snli-k5-fixed-cosine with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("swardiantara/bert-tiny-snli-k5-fixed-cosine") sentences = [ "A saddle bronc rider gets lifted out of the saddle, but keeps his grip during his ride. [SEP] The man rides an animal.", "An older man in a red vest and blue cap is walking with a cane carrying a white bag. [SEP] The man is wearing green shorts.", "A black and white dog running on the beach while a man stands behind it. [SEP] There is a dog running.", "A man on stilts is playing a tuba for money on the boardwalk. [SEP] A male street performer with a tuba is playing outside." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4]
The community tab is the place to discuss and collaborate with the HF community!
