kamangir commited on
Commit
eb26a46
1 Parent(s): 6e72b38

validating fashion_mnist train - kamangir/bolt#692

Browse files
abcli/image_classifier.sh CHANGED
@@ -71,7 +71,7 @@ function abcli_image_classifier_predict() {
71
  fi
72
 
73
  cp -v ../$data_object/*.pyndarray .
74
- cp -v ../$model_object/class_names.json .
75
 
76
  python3 -m image_classifier \
77
  predict \
 
71
  fi
72
 
73
  cp -v ../$data_object/*.pyndarray .
74
+ cp -v ../$model_object/image_classifier/model/class_names.json .
75
 
76
  python3 -m image_classifier \
77
  predict \
image_classifier/__init__.py CHANGED
@@ -1,5 +1,5 @@
1
  name = "image_classifier"
2
 
3
- version = "1.1.76"
4
 
5
  description = "fashion-mnist + hugging-face + awesome-bash-cli"
 
1
  name = "image_classifier"
2
 
3
+ version = "1.1.77"
4
 
5
  description = "fashion-mnist + hugging-face + awesome-bash-cli"
image_classifier/classes.py CHANGED
@@ -33,7 +33,9 @@ class Image_Classifier(object):
33
  }
34
 
35
  def load(self, model_path):
36
- success, self.class_names = file.load_json(f"{model_path}/class_names.json")
 
 
37
  if not success:
38
  return False
39
 
@@ -267,7 +269,7 @@ class Image_Classifier(object):
267
  self.params["model_size"] = file.size(f"{model_path}/image_classifier/model")
268
 
269
  if not file.save_json(
270
- f"{model_path}/class_names.json",
271
  self.class_names,
272
  ):
273
  return False
 
33
  }
34
 
35
  def load(self, model_path):
36
+ success, self.class_names = file.load_json(
37
+ f"{model_path}/image_classifier/model/class_names.json"
38
+ )
39
  if not success:
40
  return False
41
 
 
269
  self.params["model_size"] = file.size(f"{model_path}/image_classifier/model")
270
 
271
  if not file.save_json(
272
+ f"{model_path}/image_classifier/model/class_names.json",
273
  self.class_names,
274
  ):
275
  return False