sentence-transformers How to use Den4ikAI/rubert-tiny2-retriever with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("Den4ikAI/rubert-tiny2-retriever")
sentences = [
"query: Когда родился Пушкин?",
"passage: Алекса́ндр Серге́евич Пу́шкин (26 мая [6 июня] 1799, Москва — 29 января [10 февраля] 1837, Санкт-Петербург) — русский поэт, драматург и прозаик, заложивший основы русского реалистического направления[2], литературный критик[3] и теоретик литературы, историк[3], публицист, журналист[3].",
"passage: Пушкин ловил кайф со своими друзьями",
"passage: Пушкин из самых авторитетных литературных деятелей первой трети XIX века. Ещё при жизни Пушкина сложилась его репутация величайшего национального русского поэта[4][5]. Пушкин рассматривается как основоположник современного русского литературного языка[~ 2]."
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4]
