Merge branch 'dev/api'

Dilydigre · May 29, 2024 · 7077c35 · 7077c35
2 parents 4aa9ae7 + 8261617
commit 7077c35
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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: