use_gpu false by default

texta_mlp/mlp.py

@@ -25,7 +25,8 @@ from texta_mlp.settings import (
     SUPPORTED_ANALYZERS,
     DEFAULT_ANALYZERS,
     ENTITY_MAPPER_DATA_URLS,
-    CONCATENATOR_DATA_FILES
+    CONCATENATOR_DATA_FILES,
+    USE_GPU
 )
 
 
@@ -39,7 +40,7 @@ class MLP:
             resource_dir: str = DEFAULT_RESOURCE_DIR,
             ner_model_langs: list = CUSTOM_NER_MODEL_LANGS,
             logging_level: str ="error",
-            use_gpu: bool = True,
+            use_gpu: bool = USE_GPU,
             gpu_device_id: int = 0,
             refresh_data: bool = REFRESH_DATA
     ):

texta_mlp/settings.py

@@ -66,3 +66,6 @@ STANZA_NER_SUPPORT = ("ar", "zh", "nl", "en", "fr", "de", "ru", "es", "uk")
 
 # Here we add langs that will have custom ner models.
 CUSTOM_NER_MODEL_LANGS = ["et"]
+
+# Use gpu for pytorch
+USE_GPU = parse_bool_env("TEXTA_MLP_USE_GPU", False)