From 566a8cd868f811353d93bf9f765eb32f4832608e Mon Sep 17 00:00:00 2001 From: Jakub-Prus <68164819+Jakub-Prus@users.noreply.github.com> Date: Mon, 6 Jun 2022 23:46:27 +0200 Subject: [PATCH] small fix --- imageClasification/Classificator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)]