Update app.py

app.py

@@ -69,7 +69,8 @@ num_anchors = 9
 model = None
 
 def prepare_model(approach: int):
-    global model
+    global model, anchor_boxes, input_shape, class_names
+
     if approach not in [1, 2, 3]:
         raise ValueError("Approach must be 1, 2, or 3.")
 
@@ -80,14 +81,24 @@ def prepare_model(approach: int):
     }
     
     weight_path = os.path.join("model-data", "weights", weight_files[approach])
+    
     if not os.path.exists(weight_path):
         raise FileNotFoundError(f"Weight file not found: {weight_path}")
 
+    # Define anchor boxes for YOLO (you may need to adjust these values)
+    anchor_boxes = np.array([
+        np.array([[10, 13], [16, 30], [33, 23]]) / 32,
+        np.array([[30, 61], [62, 45], [59, 119]]) / 16,
+        np.array([[116, 90], [156, 198], [373, 326]]) / 8
+    ], dtype="float64")
+
     input_tensor = Input(shape=(input_shape[0], input_shape[1], 3))
-    num_out_filters = (num_anchors // 3) * (5 + num_classes)
+    num_out_filters = (num_anchors // 3) * (5 + len(class_names))
     model = yolo_body(input_tensor, num_out_filters)
     model.load_weights(weight_path)
 
+    print(f"YOLO model (Approach={approach}) loaded successfully.")
+
 @app.on_event("startup")
 def on_startup():
     fix_tf_gpu()