From a71f22dae54f830dbf68b3bd5e5e8d540e338a4c Mon Sep 17 00:00:00 2001
From: Rohan Anand <96521078+rohan472000@users.noreply.github.com>
Date: Thu, 30 Mar 2023 10:39:21 +0530
Subject: [PATCH] Update cnn_classification.py (#8570)

---
 computer_vision/cnn_classification.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/computer_vision/cnn_classification.py b/computer_vision/cnn_classification.py
index 1c193fcbb..9b5f8c95e 100644
--- a/computer_vision/cnn_classification.py
+++ b/computer_vision/cnn_classification.py
@@ -93,7 +93,7 @@ if __name__ == "__main__":
     test_image = tf.keras.preprocessing.image.img_to_array(test_image)
     test_image = np.expand_dims(test_image, axis=0)
     result = classifier.predict(test_image)
-    training_set.class_indices
+    # training_set.class_indices
     if result[0][0] == 0:
         prediction = "Normal"
     if result[0][0] == 1: