How to use swardiantara/bert-tiny-sst5-k1-fixed-cosine with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("swardiantara/bert-tiny-sst5-k1-fixed-cosine") sentences = [ "there 's a neat twist , subtly rendered , that could have wrapped things up at 80 minutes , but kang tacks on three or four more endings .", "such a premise is ripe for all manner of lunacy , but kaufman and gondry rarely seem sure of where it should go .", "meyjes focuses too much on max when he should be filling the screen with this tortured , dull artist and monster-in-the - making .", "director lee has a true cinematic knack , but it 's also nice to see a movie with its heart so thoroughly , unabashedly on its sleeve ." ] 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!
