good sample to no defect

scripts/welspun_classifier.py

@@ -225,7 +225,7 @@ class Welspun_Classifier(ModelWrapper):
         # if self.counter%1 == 0:
         self.counter = self.counter + 1
 
-        if a == 0 and max(x) > 0:
+        if a == 0 and x[0] > 0.95:
             resized_frame = cv2.resize(frame, (64, 64))
             cv2.putText(frame, text="Mix Color Defect Detected", org=(50, 50),
                         color=(0, 0, 255),
@@ -238,7 +238,7 @@ class Welspun_Classifier(ModelWrapper):
                                   "sound_1")
                 logger.info(f"Probability: {prob}")
                 self.counter = 0
-        elif a == 2 and max(x) > 0:
+        elif a == 2 and x[2] > 0.95:
             resized_frame = cv2.resize(frame, (64, 64))
             cv2.putText(frame, text="Split Defect Detected", org=(50, 50),
                         color=(0, 0, 255),
@@ -250,7 +250,7 @@ class Welspun_Classifier(ModelWrapper):
                                   "sound_1")
                 logger.info(f"Probability: {prob}")
                 self.counter = 0
-        elif a == 1 and max(x) > 0:
+        elif a == 1 and x[1] > 0.95:
             resized_frame = cv2.resize(frame, (64, 64))
             cv2.putText(frame, text="Short Tile Defect Detected", org=(50, 50),
                         color=(0, 0, 255),