model.py

import keras_cv
import tensorflow as tf


backbone= keras_cv.models.YOLOV8Backbone.from_preset(
    "yolo_v8_m_backbone_coco",
    load_weights=False
)
model = keras_cv.models.YOLOV8Detector(
    num_classes=2,
    bounding_box_format="xyxy",
    backbone=backbone,
    fpn_depth=2,
)

model.load_weights('models/model.weights.h5')