use_gpu env variable

worker/settings.py

@@ -8,3 +8,4 @@ MLP_WORKER_DEFAULT_LANGUAGE_CODE = os.getenv("MLP_WORKER_DEFAULT_LANGUAGE_CODES"
 MLP_WORKER_RESOURCE_DIR = os.getenv("MLP_WORKER_RESOURCE_DIR", "../data")
 MLP_WORKER_BROKER = os.getenv("MLP_WORKER_REDIS_URL", "redis://mlp-redis:6379/1")
 MLP_WORKER_RESULT_BACKEND = os.getenv("MLP_WORKER_REDIS_RESULT_BACKEND", MLP_WORKER_BROKER)
+MLP_USE_GPU = os.getenv("MLP_USE_GPU", "true")

worker/taskman.py

@@ -4,7 +4,7 @@ from typing import List, Optional
 from celery import Celery, shared_task
 
 from mlp_docparser import MLPDocParser
-from settings import (MLP_WORKER_BROKER, MLP_WORKER_DEFAULT_LANGUAGE_CODE, MLP_WORKER_LANGUAGE_CODES, MLP_WORKER_RESOURCE_DIR, MLP_WORKER_RESULT_BACKEND)
+from settings import (MLP_WORKER_BROKER, MLP_WORKER_DEFAULT_LANGUAGE_CODE, MLP_WORKER_LANGUAGE_CODES, MLP_WORKER_RESOURCE_DIR, MLP_WORKER_RESULT_BACKEND, MLP_USE_GPU)
 from texta_mlp.mlp import MLP
 
 
@@ -31,7 +31,8 @@ def load_mlp():
         ml_processor = MLP(
             language_codes=MLP_WORKER_LANGUAGE_CODES,
             default_language_code=MLP_WORKER_DEFAULT_LANGUAGE_CODE,
-            resource_dir=MLP_WORKER_RESOURCE_DIR
+            resource_dir=MLP_WORKER_RESOURCE_DIR,
+            use_gpu=MLP_USE_GPU
         )
         logging.info("Successfully loaded MLP models.")