b2u commited on
Commit
f85629e
·
1 Parent(s): a3aa3ec

adding model directory

Browse files
Files changed (1) hide show
  1. model.py +4 -0
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)