behnamsa commited on
Commit
2167368
1 Parent(s): dc5b7ba

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +3 -3
pipeline.py CHANGED
@@ -11,9 +11,9 @@ from scipy.spatial import distance
11
  class PreTrainedPipeline():
12
  def __init__(self, path):
13
  self.model_dir = path + "/saved_model"
14
- self.t2id_path = path + "t2id.json"
15
- self.stopwords_path = path + "stopwords.txt"
16
- self.id2h_path = path + "id2h.json"
17
  self.t2id = json.load(open(self.t2id_path,encoding="utf8"))
18
  self.id2h = json.load(open(self.id2h_path,encoding="utf8"))
19
 
 
11
  class PreTrainedPipeline():
12
  def __init__(self, path):
13
  self.model_dir = path + "/saved_model"
14
+ self.t2id_path = path + "/t2id.json"
15
+ self.stopwords_path = path + "/stopwords.txt"
16
+ self.id2h_path = path + "/id2h.json"
17
  self.t2id = json.load(open(self.t2id_path,encoding="utf8"))
18
  self.id2h = json.load(open(self.id2h_path,encoding="utf8"))
19