small fix

2022-06-06 23:46:27 +02:00 · 2022-06-06 23:46:27 +02:00 · 566a8cd868
commit 566a8cd868
parent e5a7a975e8
1 changed files with 1 additions and 1 deletions
--- 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)]