model-index:
  - name: Rock vs Non-Rock Model
    task:
      type: object-detection
    framework: pytorch
    tags:
      - yolov8
      - rock-classification
    config: config.yaml
    weights: best.pt
    labels:
      0: rock
      1: non-rock