Merge pull request #4 from atrifat/fix-language-detection-model-multi…

…ple-download feat: Warming up language detection model before starting API server
atrifat · Oct 11, 2024 · d25e7c4 · d25e7c4
2 parents ed0bd65 + 92408a5
commit d25e7c4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/app.py b/app.py
@@ -49,6 +49,10 @@
 if ENABLE_API_TOKEN and API_TOKEN == "":
     raise Exception("API_TOKEN is required if ENABLE_API_TOKEN is enabled")
 
+#  Ensure model warmed-up for fast_langdetect before starting API server
+if LANGUAGE_DETECTION_MODEL == "fast_langdetect":
+    detect_multilingual("", low_memory=LOW_MEMORY_MODE)
+
 app = Flask(__name__)
 
 cache_config = {