main.py 551 B

123456789101112131415161718192021
  1. from fastapi import FastAPI
  2. from contextlib import asynccontextmanager
  3. from .core.model_loader import load_models, unload_models
  4. from app.api.card_inference import router as card_inference_router
  5. from .core.config import settings
  6. @asynccontextmanager
  7. async def lifespan(app: FastAPI):
  8. print("--- 应用启动 ---")
  9. load_models()
  10. yield
  11. print("--- 应用关闭 ---")
  12. unload_models()
  13. app = FastAPI(title="卡片框和缺陷检测服务", lifespan=lifespan)
  14. app.include_router(card_inference_router, prefix=settings.API_prefix)