diff --git a/app/main.py b/app/main.py index 7b68ed6..c1e4934 100644 --- a/app/main.py +++ b/app/main.py @@ -3,11 +3,14 @@ from .routers import model_v2 from .routers import response_model from .config import CONFIG +import os app = FastAPI() app.include_router(model_v1.router.router) # router for model 1 -app.include_router(model_v2.router.router) # router for model 2 + +if os.environ['INCLUDE_V2'] == 'true': + app.include_router(model_v2.router.router) # router for model 2 @app.get("/", response_model=response_model.StatusResponse) async def root() -> response_model.StatusResponse: