validating single image predict for fashion_mnist - kamangir/bolt#692

image_classifier/__init__.py

@@ -1,5 +1,5 @@
 name = "image_classifier"
 
-version = "1.1.149"
+version = "1.1.151"
 
 description = "fashion-mnist + hugging-face + awesome-bash-cli"

image_classifier/classes.py

@@ -314,6 +314,7 @@ class Image_Classifier(object):
     @staticmethod
     def train(data_path, model_path, color=False, convnet=True, epochs=10):
         classifier = Image_Classifier()
+        classifier.params["color"] = color
         classifier.params["convnet"] = convnet
 
         logger.info(