added image ai-moderation
This commit is contained in:
@@ -1,11 +1,15 @@
|
||||
import os
|
||||
from contextlib import asynccontextmanager
|
||||
from fastapi import FastAPI
|
||||
from app.ml.model_manager import model_manager
|
||||
from app.config.logging import logger
|
||||
|
||||
from app.config.settings import settings
|
||||
|
||||
@asynccontextmanager
|
||||
async def lifespan(app: FastAPI):
|
||||
if settings.HF_TOKEN:
|
||||
os.environ["HF_TOKEN"] = settings.HF_TOKEN
|
||||
|
||||
logger.info("Loading AI models...")
|
||||
model_manager.initialize()
|
||||
logger.info("AI models loaded.")
|
||||
|
||||
Reference in New Issue
Block a user