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

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +4 -4
pipeline.py CHANGED
@@ -10,10 +10,10 @@ from scipy.spatial import distance
10
 
11
  class PreTrainedPipeline():
12
  def __init__(self, path):
13
- self.model_dir = "saved_model"
14
- self.t2id_path = "t2id.json"
15
- self.stopwords_path = "stopwords.txt"
16
- self.id2h_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
 
 
10
 
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