diff --git a/fastNLP/api/api.py b/fastNLP/api/api.py index 202f782f..b557038b 100644 --- a/fastNLP/api/api.py +++ b/fastNLP/api/api.py @@ -1,11 +1,16 @@ +import _pickle + class API: def __init__(self): - pass + self.pipeline = None + self.model = None def predict(self): pass - def load(self): - pass \ No newline at end of file + def load(self, name): + _dict = _pickle.load(name) + self.pipeline = _dict['pipeline'] + self.model = _dict['model']