Browse Source

disable revalidation if model unloading disabled

mertalev 2 years ago
parent
commit
7d44deccc5
1 changed files with 1 additions and 1 deletions
  1. 1 1
      machine-learning/app/main.py

+ 1 - 1
machine-learning/app/main.py

@@ -25,7 +25,7 @@ app = FastAPI()
 
 
 def init_state() -> None:
-    app.state.model_cache = ModelCache(ttl=settings.model_ttl, revalidate=True)
+    app.state.model_cache = ModelCache(ttl=settings.model_ttl, revalidate=settings.model_ttl > 0)
 
 
 async def load_models() -> None: