andrewrreed HF staff commited on
Commit
ac70c4f
1 Parent(s): 1572885

fix data dir

Browse files
Files changed (1) hide show
  1. handler.py +3 -2
handler.py CHANGED
@@ -8,8 +8,9 @@ class EndpointHandler:
8
  def __init__(self, path=""):
9
  self.model = GECToR.from_pretrained(path)
10
  self.tokenizer = AutoTokenizer.from_pretrained(path)
11
- print(os.getcwd())
12
- self.encode, self.decode = load_verb_dict("./data/verb-form-vocab.txt")
 
13
 
14
  def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
15
  """
 
8
  def __init__(self, path=""):
9
  self.model = GECToR.from_pretrained(path)
10
  self.tokenizer = AutoTokenizer.from_pretrained(path)
11
+ self.encode, self.decode = load_verb_dict(
12
+ os.path.join(path, "data/verb-form-vocab.txt")
13
+ )
14
 
15
  def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
16
  """