|
@@ -1,5 +1,5 @@ |
|
|
|
|
|
|
|
|
import _pickle |
|
|
|
|
|
|
|
|
import torch |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class API: |
|
|
class API: |
|
@@ -11,6 +11,6 @@ class API: |
|
|
pass |
|
|
pass |
|
|
|
|
|
|
|
|
def load(self, name): |
|
|
def load(self, name): |
|
|
_dict = _pickle.load(name) |
|
|
|
|
|
|
|
|
_dict = torch.load(name) |
|
|
self.pipeline = _dict['pipeline'] |
|
|
self.pipeline = _dict['pipeline'] |
|
|
self.model = _dict['model'] |
|
|
self.model = _dict['model'] |