Commit
·
1572885
1
Parent(s):
856f178
debug data dir
Browse files- 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]]:
|