Valeriy Sinyukov
commited on
Commit
·
bf2b17b
1
Parent(s):
d8f96d2
Remove unused get_state method
Browse files
category_classification/models/oracat__bert_paper_classifier/model.py
CHANGED
|
@@ -10,9 +10,6 @@ class BertPaperClassifierModel:
|
|
| 10 |
def __call__(self, input):
|
| 11 |
return self.pipeline(input.title + ' ' + input.abstract)
|
| 12 |
|
| 13 |
-
def __getstate__(self):
|
| 14 |
-
return self.__dict__
|
| 15 |
-
|
| 16 |
def get_model():
|
| 17 |
return BertPaperClassifierModel()
|
| 18 |
|
|
|
|
| 10 |
def __call__(self, input):
|
| 11 |
return self.pipeline(input.title + ' ' + input.abstract)
|
| 12 |
|
|
|
|
|
|
|
|
|
|
| 13 |
def get_model():
|
| 14 |
return BertPaperClassifierModel()
|
| 15 |
|