Captioning-bot

Telegram-бот с нейронной сетью для генерации описаний к изображениям (image captioning).

Для запуска проекта необходимо:

• Установить Python 3.9
• Клонировать репозиторий и установить зависимости:
  • git clone https://github.com/PandaMia/Captioning-bot.git
  • cd Captioning-bot
  • pip install -r requirements.txt
• Скачать веса для модели генерации описаний и разместить по пути source/model:
  • Веса модели
• Зарегистрировать своего телеграм-бота через @BotFather. Полученный токен бота присвоить в переменную TOKEN в файле bot.py.
• В консоли вызвать:
  • python bot.py

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