/neuro_writer

Neural network for generate text. Used russian GPT-2 from https://github.com/sberbank-ai/ru-gpts. The model was trained on the works of Mayakovsky.

Primary LanguageJupyter NotebookOtherNOASSERTION

Neuro_mayakovsky

Language model GPT-2 for generate text in style of V. V. Maykovsky

About project

Used Russian GPT-2 from: https://github.com/sberbank-ai/ru-gpts

The collected works of V. V. Mayakovsky are used as data

Model

Copy files of checkpoint from https://drive.google.com/drive/folders/17uvNjAezqwgfWefrHijpx__zznt4kDzY?usp=sharing to your google drive

Using model for generation text

See /notebook/generation.ipynb

Training the model on your data

See /notebook/train_model.ipynb

Examples of generated texts

You can see examples in instagram @neuro_mayakovsky:

https://www.instagram.com/neuro_mayakovsky/