Spaces:
Sleeping
Sleeping
adding model directory
Browse files
model.py
CHANGED
|
@@ -35,6 +35,10 @@ class BertClassifier(LabelStudioMLBase):
|
|
| 35 |
def __init__(self, project_id=None, label_config=None, **kwargs):
|
| 36 |
super(BertClassifier, self).__init__(project_id=project_id, label_config=label_config)
|
| 37 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 38 |
# Parse label config to get categories
|
| 39 |
from label_studio_ml.model import parse_config
|
| 40 |
parsed_config = parse_config(label_config)
|
|
|
|
| 35 |
def __init__(self, project_id=None, label_config=None, **kwargs):
|
| 36 |
super(BertClassifier, self).__init__(project_id=project_id, label_config=label_config)
|
| 37 |
|
| 38 |
+
# Set up model directory
|
| 39 |
+
self.model_dir = os.path.join(os.path.dirname(__file__), 'model')
|
| 40 |
+
os.makedirs(self.model_dir, exist_ok=True)
|
| 41 |
+
|
| 42 |
# Parse label config to get categories
|
| 43 |
from label_studio_ml.model import parse_config
|
| 44 |
parsed_config = parse_config(label_config)
|