modified all

app.py

@@ -13,7 +13,7 @@ app = Flask(__name__)
 trained_model = YOLO("best.pt")
 
 # Load TFLite models
-sp_interpreter = tf.lite.Interpreter(model_path='small_picture.tflite')
+sp_interpreter = tf.lite.Interpreter(model_path='flag.tflite')
 sp_interpreter.allocate_tensors()
 
 c_interpreter = tf.lite.Interpreter(model_path='model_chip_v4.tflite')
@@ -88,8 +88,8 @@ def predict():
                         confidence_score_c = float(output_data_c[0][0])
                         # print("Confidence Score C:", confidence_score_c)
 
-                    elif class_id == 3:
-                        cropped_img = cv2.resize(cropped_img, (128, 128))
+                    elif class_id == 1:
+                        cropped_img = cv2.resize(cropped_img, (92, 92))
                         cropped_img = cropped_img.astype('float32') / 255.0
                         cropped_img -= imagenet_mean
                         cropped_img /= imagenet_std

flag.tflite

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a998b77d5f0158cc0c07f13582bd8b7cdf33c28490f4f77ef38129d3bed92b95
+size 3416328

templates/index.html

@@ -340,7 +340,7 @@
                 confidenceScore.textContent =
                   "Confidence Score Chip: " +
                   data.confidence_score_c +
-                  "Confidence Score Small Picture: " +
+                  "Confidence Score Flag: " +
                   data.confidence_score_sp;
                 uploadSuccess.style.display = "block"; // Show success message
               } else if (data.confidence_score_c !== undefined) {
@@ -351,7 +351,7 @@
               } else if (data.confidence_score_sp !== undefined) {
                 resultsImg.src = canvas.toDataURL("image/png"); // Display captured image
                 confidenceScore.textContent =
-                  "Confidence Score Small Picture: " + data.confidence_score_sp;
+                  "Confidence Score Flag: " + data.confidence_score_sp;
                 uploadSuccess.style.display = "block"; // Show success message
               } else {
                 alert(data.error);
@@ -400,7 +400,7 @@
                 confidenceScore.textContent =
                   "Confidence Score Chip: " +
                   data.confidence_score_c +
-                  "\nConfidence Score Small Picture: " +
+                  "\nConfidence Score Flag : " +
                   data.confidence_score_sp;
                 uploadSuccess.style.display = "block"; // Show success message
               };
@@ -416,7 +416,7 @@
               reader.readAsDataURL(file);
             } else if (data.confidence_score_sp !== undefined) {
               confidenceScore.textContent =
-                "Confidence Score Small Picture: " + data.confidence_score_sp;
+                "Confidence Score Flag: " + data.confidence_score_sp;
             } else {
               console.error("Error: Confidence score is undefined.");
               alert(data.error || "An unexpected error occurred.");