Тг бот, меняющий стиль фотографий с помощью нейросети

Установка

Скачайте код, и перейдите в каталог проекта. Затем создайте файл .env и положите в него следующие параметры в формате Ключ=значение:

-TG_BOT_TOKEN - Токен вашего бота
-IMAGE_SIZE - Размер изображений, которые ваш бот будет генерировать

По умолчанию IMAGE_SIZE= 256, при отсутствии GPU с таким значением генерация изображения займёт около минуты.

Установите необходимые зависимости:

pip3 install -r requirements.txt

Далее вам понадобится установить PyTorch. Его метод установки зависит от вашей OC, а также конфигурации устройства, поэтому он не включем в requirements.txt. Перейдите на официальный сайт и проследуйте их инструкциям.

Запуск

python tg_bot.py 

Первый запуск займёт чуть больше времени, так как скрипт будет загружать необходимые веса для модели.

Пример работы:

Исходное изображение

Стиль

Результат

Источник картинки