GUI-программа для описания картинок на основе моделей FancyFeast и Florence-2, на движке HuggingFace.
- менеджмент картинок (добавление, удаление, сохранение в список, загрузка списка и.т.д.)
- AI описание списка
- Описание отдельных картинок
- 3 подключенные модели: FancyFeast, Florence-2, Florence SD3
- Возможность полного, краткого и полного + краткого описания картинок за счет модели Bart от Facebook
- Два языка: русский и англиский
- Добавление ключевых слов во все списки (в начале, в конце, замена)
- автокопирование со сдвигом по горячей клавише F12
- сохранение картинок с описанием в Excel документ
Одна из важных возможностей программы - копировать текст из полей по горячей клавише F12 + вставка в любое текстовое поле. Благодаря этому можно заполнять поля, не переключаясь в программу и не отслеживая какая фотография по счету идет
Программа собрана на PyQt5 с подключением моделей FancyFeast и Florence-2 на движке HuggingFace.
Чтобы собрать программу, используйте эту команду
pyinstaller main.spec
Для бОльших возможностей - нужно создать и подключить токен huggingface (https://huggingface.co/settings/tokens) в программе - бОльшее количество генераций в сутки
FancyFeast: https://huggingface.co/spaces/fancyfeast/joy-caption-pre-alpha
Florence-2: https://huggingface.co/spaces/gokaygokay/Florence-2
Florence-SD3: https://huggingface.co/spaces/gokaygokay/Florence-2-SD3-Captioner
Bart: https://huggingface.co/facebook/bart-large-cnn