sentence-transformers How to use abragin/ruBert-style-base with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("abragin/ruBert-style-base")
sentences = [
"Шатов вдруг прокричал кратким и отчаянным криком; но ему кричать не дали: Петр Степанович аккуратно и твердо наставил ему револьвер прямо в лоб, крепко в упор и – спустил курок. Выстрел, кажется, был не очень громок, по крайней мере в Скворешниках ничего не слыхали. Слышал, разумеется, Шигалев, вряд ли успевший отойти шагов триста, – слышал и крик и выстрел, но, по его собственному потом свидетельству, не повернулся и даже не остановился. Смерть произошла почти мгновенно. Полную распорядительность – не думаю, чтоб и хладнокровие, – сохранил в себе один только Петр Степанович. Присев на корточки, он поспешно, но твердою рукой обыскал в карманах убитого. Денег не оказалось (портмоне остался под подушкой у Марьи Игнатьевны). Нашлись две-три бумажки, пустые: одна конторская записка, заглавие какой-то книги и один старый заграничный трактирный счет, бог знает почему уцелевший два года в его кармане. Бумажки Петр Степанович переложил в свой карман и, заметив вдруг, что все столпились, смотрят на труп и ничего не делают, начал злостно и невежливо браниться и понукать. Толкаченко и Эркель, опомнившись, побежали и мигом принесли из грота еще с утра запасенные ими там два камня, каждый фунтов по двадцати весу, уже приготовленные, то есть крепко и прочно обвязанные веревками.",
"Черт таким же порядком отправился вслед за нею. Но так как это животное проворнее всякого франта в чулках, то не мудрено, что он наехал при самом входе в трубу на шею своей любовницы, и оба очутились в просторной печке между горшками. Путешественница отодвинула потихоньку заслонку, поглядеть, не назвал ли сын ее Вакула в хату гостей, но, увидевши, что никого не было, выключая только мешки, которые лежали посреди хаты, вылезла из печки, скинула теплый кожух, оправилась, и никто бы не мог узнать, что она за минуту назад ездила на метле. Мать кузнеца Вакулы имела от роду не больше сорока лет. Она была ни хороша, ни дурна собою. Трудно и быть хорошею в такие годы. Однако ж она так умела причаровать к себе самых степенных козаков (которым, не мешает, между прочим, заметить, мало было нужды до красоты), что к ней хаживал и голова, и дьяк Осип Никифорович (конечно, если дьячихи не было дома), и козак Корний Чуб, и козак Касьян Свербыгуз. И, к чести ее сказать, она умела искусно обходиться с ними. Ни одному из них и в ум не приходило, что у него есть соперник.",
"Поднимаясь в свою квартиру, он заметил, что Настасья, оторвавшись от самовара, пристально следит за ним и провожает его глазами. «Уж нет ли кого у меня?» – подумал он. Ему с отвращением померещился Порфирий. Но, дойдя до своей комнаты и отворив ее, он увидел Дунечку. Она сидела одна-одинешенька, в глубоком раздумье и, кажется, давно уже ждала его. Он остановился на пороге. Она привстала с дивана в испуге и выпрямилась пред ним. Ее взгляд, неподвижно устремленный на него, изображал ужас и неутолимую скорбь.",
"Девушки между тем, дружно взявшись за руки, полетели, как вихорь, с санками по скрыпучему снегу. Множество, шаля, садились на санки; другие взбирались на самого голову. Голова решился сносить все. Наконец приехали, отворили настежь двери в сенях и хате и с хохотом втащили мешок. — Посмотрим, что-то лежит тут, — закричали все, бросившись развязывать. Тут икотка, которая не переставала мучить голову во все время сидения его в мешке, так усилилась, что он начал икать и кашлять во все горло. — Ах, тут сидит кто-то!"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4] 
