diff --git a/imageClasification/Classificator.py b/imageClasification/Classificator.py index f50f2fa..c7e2ea6 100644 --- a/imageClasification/Classificator.py +++ b/imageClasification/Classificator.py @@ -16,7 +16,7 @@ def image_classification(path, model): predictions = model.predict(img_array) score = tf.nn.softmax(predictions[0]) - print(class_names[np.argmax(score)]) + # print(class_names[np.argmax(score)]) return class_names[np.argmax(score)]