Merge pull request #151 from NervanaSystems/docker-fix

Fixed server modules

Author: Peter Izsak

nlp_architect/api/intent_extraction_api.py

@@ -30,7 +30,7 @@ class IntentExtractionApi(AbstractApi):
     pretrained_model_info = path.join(model_dir, "model_info.dat")
     pretrained_model = path.join(model_dir, "model.h5")
 
-    def __init__(self, prompt=True):
+    def __init__(self, prompt=False):
         self.model = None
         self.model_type = None
         self.word_vocab = None

nlp_architect/api/ner_api.py

@@ -35,7 +35,7 @@ class NerApi(AbstractApi):
     pretrained_model = path.join(model_dir, "model_v4.h5")
     pretrained_model_info = path.join(model_dir, "model_info_v4.dat")
 
-    def __init__(self, prompt=True):
+    def __init__(self, prompt=False):
         self.model = None
         self.model_info = None
         self.word_vocab = None

server/README.md

@@ -1,7 +1,7 @@
 # NLP Architect Server
 
 ## Running the server
-1. `hug -p 8080 server/server.py # from root dir`
+1. `hug -p 8080 -f server/server.py # from root dir`
 
 ## Development
 1. Install Node.js from [here](https://nodejs.org/en/)

server/serve.py

@@ -31,7 +31,7 @@
 
 
 def prefetch_models():
-    models = ['machine_comprehension', 'bist', 'ner', 'intent_extraction']
+    models = ['bist', 'ner', 'intent_extraction']
     for model in models:
         services[model] = Service(model)
 

server/services.json

@@ -2,6 +2,5 @@
   "api_folders_path" : "nlp_architect.api.",
   "bist" : {"file_name": "bist_parser_api.py", "type": "core", "spacy": true},
   "ner" : {"file_name": "ner_api.py", "type": "high_level"},
-  "machine_comprehension" : {"file_name": "machine_comprehension_api.py", "type": "high_level"},
   "intent_extraction" : {"file_name": "intent_extraction_api.py", "type": "high_level"}
-}
+}