Сервис для вызова модели.
Создайте и активируйте виртуальное окружение. Установите зависимости
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
возвращал путь до сгенерированной картинки. Если модель возвращает что-то другое надо будет обновить эндпоинт.