kamangir
commited on
Commit
•
3022254
1
Parent(s):
958fcc3
saving fashion_mnist - kamangir/bolt#692
Browse files- abcli/fashion_mnist.sh +4 -24
- abcli/image_classifier.sh +8 -0
- image_classifier/__init__.py +1 -1
abcli/fashion_mnist.sh
CHANGED
@@ -35,30 +35,10 @@ function abcli_fashion_mnist() {
|
|
35 |
fi
|
36 |
|
37 |
if [ "$task" == "predict" ] ; then
|
38 |
-
|
39 |
-
|
40 |
-
|
41 |
-
|
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.
|
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"
|