Use MODEL env var

app/src/tasks.py

@@ -3,6 +3,7 @@
 """
 
 from typing import Any
+import os
 
 from celery import Task
 from loguru import logger
@@ -27,7 +28,7 @@ def __call__(self, *args, **kwargs) -> Any:
         if not self.model:
             # Protecting this in the __call__ method to avoid
             # loading the model in the fastapi server process.
-            self.model = NERModel()
+            self.model = NERModel(os.getenv("MODEL"))
         return self.run(*args, **kwargs)