updated infer.py and project config for new NER class

project/msforum_xx.config.json

@@ -26,7 +26,8 @@
         },
         "3": {
             "type": "ner",
-            "prepare": false
+            "prepare": false,
+            "models": ["textanalytics", "flair", "custom", "regex", "nerlist"]
         },
         "4": {
             "type": "qa",

src/infer.py

@@ -32,7 +32,7 @@ def score(task):
         _dt = dt.Data(task=task, inference=True)
         return Inferencer.load(_dt.get_path('model_dir'))   
     elif task_type == 'ner':
-        return ner.NER(task=task, inference=True)
+        return ner.NER(task=task, inference=True, models=cu.tasks.get(str(task)).get('models'))
     elif task_type == 'qa':
         return rank.Rank(task=task, inference=True)
     elif task_type == 'om':