Update app.py

app.py

@@ -69,8 +69,7 @@ num_anchors = 9
 model = None
 
 def prepare_model(approach: int):
-    global model, anchor_boxes, input_shape, class_names
-
+    global model
     if approach not in [1, 2, 3]:
         raise ValueError("Approach must be 1, 2, or 3.")
 
@@ -81,24 +80,14 @@ 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 + len(class_names))
+    num_out_filters = (num_anchors // 3) * (5 + num_classes)
     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()