sentence-transformers How to use PyxiLab/Pyx-embeds with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("PyxiLab/Pyx-embeds")
sentences = [
"search_query: i love autotrain",
"search_query: huggingface auto train",
"search_query: hugging face auto train",
"search_query: i love autotrain"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4] 
