Update epi_pipeline.py

epi_pipeline.py

@@ -231,13 +231,13 @@ import spacy
 import json
 
 class Classify_Pipeline:
-    def __init__(self,model:str='LSTM_RNN_Model'):
+    def __init__(self,model_name:str='LSTM_RNN_Model'):
         #Load spaCy models
         self.nlp = spacy.load('en_core_web_lg')
         self.nlpSci = spacy.load("en_ner_bc5cdr_md")
         self.nlpSci2 = spacy.load('en_ner_bionlp13cg_md')
         # load the tokenizer
-        with open('tokenizer.json') as f:
+        with open(model_name+'/tokenizer.json') as f:
             self.classify_tokenizer = tokenizer_from_json(json.load(f))
         #OLD Code - used pickle which is unsafe
         #with open(model+'/tokenizer.pickle', 'rb') as handle:
@@ -247,7 +247,7 @@ class Classify_Pipeline:
         # but can also be run on any other tf.keras model
     
         # load the model
-        self.classify_model = tf.keras.models.load_model(model)
+        self.classify_model = tf.keras.models.load_model(model_name)
         # for preprocessing
         from nltk.corpus import stopwords
         self.STOPWORDS = set(stopwords.words('english'))