geniai

Сервис для вызова модели.

Установка

Создайте и активируйте виртуальное окружение. Установите зависимости

pip install -r requirements.txt

Запуск

cd src
uvicorn main:app --host 0.0.0.0 --port 80 --reload

Для корректной работы требуется в файле maag_geniai_model.py указать путь к весам кастомной/предобученной модели

Тогда после ввода промпта, модель сгенерирует картинку (потребуется немного врмени)

Доработка

В папку src/model положите модуль с моделью.

Пример модуля в файле src/model/generate.py. Важно, чтобы вызываемый в эндпоинте /model возвращал путь до сгенерированной картинки. Если модель возвращает что-то другое надо будет обновить эндпоинт.