Update pipeline.py

pipeline.py

@@ -12,8 +12,10 @@ class PreTrainedPipeline():
     def __init__(self, path):
         self.model_dir = path + "/saved_model"
         self.t2id_path = path + "/t2id.json"
-        self.stopwords_path = path + "/stopwords.txt"
         self.id2h_path = path + "/id2h.json"
+        self.stopwords_path = path + "/stopwords.txt"
+        self.comparison_matrix_path = path + "/comparison_matrix.npz"
+        
         self.t2id = json.load(open(self.t2id_path,encoding="utf8"))
         self.id2h = json.load(open(self.id2h_path,encoding="utf8"))