sentence-transformers How to use aihello/keyword-recommendation-beta with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("aihello/keyword-recommendation-beta")
sentences = [
"great walks uk book",
"loose match",
"loose match",
"lamina mapa montevideo"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4]