andrewrreed HF staff commited on
Commit
1572885
·
1 Parent(s): 856f178

debug data dir

Browse files
Files changed (1) hide show
  1. handler.py +2 -0
handler.py CHANGED
@@ -1,3 +1,4 @@
 
1
  from typing import Dict, List, Any
2
  from transformers import AutoTokenizer
3
  from gector import GECToR, predict, load_verb_dict
@@ -7,6 +8,7 @@ class EndpointHandler:
7
  def __init__(self, path=""):
8
  self.model = GECToR.from_pretrained(path)
9
  self.tokenizer = AutoTokenizer.from_pretrained(path)
 
10
  self.encode, self.decode = load_verb_dict("./data/verb-form-vocab.txt")
11
 
12
  def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
 
1
+ import os
2
  from typing import Dict, List, Any
3
  from transformers import AutoTokenizer
4
  from gector import GECToR, predict, load_verb_dict
 
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]]: