26 lines
789 B
Python
26 lines
789 B
Python
|
import logging
|
|||
|
|
import os
|
|||
|
|
|
|||
|
|
import litserve as ls
|
|||
|
|
|
|||
|
|
from app.config.config import settings
|
|||
|
|
from logging_env import LOGGER_CONFIG_DICT
|
|||
|
|
from app.server.generate_image.flux2_klein.server import Flux2KleinServer
|
|||
|
|
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
# 判断目录是否存在
|
|||
|
|
logs = 'logs'
|
|||
|
|
if not os.path.exists(logs):
|
|||
|
|
# 不存在则创建目录（parents=True 允许创建多级目录，exist_ok=True 避免目录已存在时报错）
|
|||
|
|
os.makedirs(logs, exist_ok=True)
|
|||
|
|
logger.info(f"目录 {logs} 创建成功")
|
|||
|
|
else:
|
|||
|
|
logger.info(f"目录 {logs} 已存在")
|
|||
|
|
logging.config.dictConfig(LOGGER_CONFIG_DICT)
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
to_product = Flux2KleinServer(api_path='/api/v1/to_product')
|
|||
|
|
|
|||
|
|
server = ls.LitServer([to_product])
|
|||
|
|
server.run(port=settings.SERVE_PROD)
|