14 lines
363 B
Python
14 lines
363 B
Python
from contextlib import asynccontextmanager
|
|
from fastapi import FastAPI
|
|
from app.ml.model_manager import model_manager
|
|
from app.config.logging import logger
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
logger.info("Loading AI models...")
|
|
model_manager.initialize()
|
|
logger.info("AI models loaded.")
|
|
yield
|
|
logger.info("Stopping AI Service...")
|