kamangir commited on
Commit
3022254
1 Parent(s): 958fcc3

saving fashion_mnist - kamangir/bolt#692

Browse files
abcli/fashion_mnist.sh CHANGED
@@ -35,30 +35,10 @@ function abcli_fashion_mnist() {
35
  fi
36
 
37
  if [ "$task" == "predict" ] ; then
38
- local data_object=$(abcli_clarify_object "$2" $abcli_object_name)
39
-
40
- local options=$4
41
- local model_is_object=$(abcli_option_get_unpacked "$options" "object" 0)
42
-
43
- if [ "$model_is_object" == 1 ] ; then
44
- local model_object=$(abcli_clarify_object "$3")
45
-
46
- abcli_download object $model_object
47
-
48
- local model_path=$abcli_object_root/$model_object
49
- else
50
- local model_name=$(abcli_clarify_arg "$3" fashion-mnist)
51
-
52
- local model_path=$abcli_path_git/image-classifier/saved_model/$model_name
53
- fi
54
-
55
-
56
- # object_1 [name_1] [object]
57
-
58
-
59
-
60
-
61
- abcli_image_classifier_predict ${@:2}
62
  return
63
  fi
64
 
 
35
  fi
36
 
37
  if [ "$task" == "predict" ] ; then
38
+ abcli_huggingface predict \
39
+ image-classifier \
40
+ $(abcli_clarify_arg "$2" fashion-mnist) \
41
+ ${@:3}
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
42
  return
43
  fi
44
 
abcli/image_classifier.sh CHANGED
@@ -48,6 +48,14 @@ function abcli_image_classifier() {
48
  return
49
  fi
50
 
 
 
 
 
 
 
 
 
51
  if [ "$task" == "save" ] ; then
52
  abcli_huggingface save \
53
  image-classifier \
 
48
  return
49
  fi
50
 
51
+ if [ "$task" == "predict" ] ; then
52
+ abcli_huggingface predict \
53
+ image-classifier \
54
+ $(abcli_clarify_arg "$2" image-classifier) \
55
+ ${@:3}
56
+ return
57
+ fi
58
+
59
  if [ "$task" == "save" ] ; then
60
  abcli_huggingface save \
61
  image-classifier \
image_classifier/__init__.py CHANGED
@@ -1,5 +1,5 @@
1
  name = "image_classifier"
2
 
3
- version = "1.1.98"
4
 
5
  description = "fashion-mnist + hugging-face + awesome-bash-cli"
 
1
  name = "image_classifier"
2
 
3
+ version = "1.1.99"
4
 
5
  description = "fashion-mnist + hugging-face + awesome-bash-cli"