Commit
•
ac70c4f
1
Parent(s):
1572885
fix data dir
Browse files- 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 |
-
|
12 |
-
|
|
|
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 |
"""
|