metadata:
  name: ultralytics-yolov5_firesmoke_1.0
  namespace: cvat
  annotations:
    name: YOLO v5_firesmoke_1.0
    type: detector
    framework: pytorch
    spec: |
      [
        { "id": 0, "name": "fire" },
        { "id": 1, "name": "smoke"}
      ]

spec:
  description: YOLO v5 via pytorch hub_firesmoke_1.0
  runtime: 'python:3.6'
  handler: main:handler
  eventTimeout: 30s
  build:
    image: cvat.ultralytics-yolov5_firesmoke_1.0
    baseImage: ultralytics/yolov5:latest-cpu

    directives:
      preCopy:
        - kind: USER
          value: root
        - kind: RUN
          value: apt update && apt install --no-install-recommends -y libglib2.0-0
        - kind: WORKDIR
          value: /opt/nuclio

  triggers:
    myHttpTrigger:
      maxWorkers: 1
      kind: 'http'
      workerAvailabilityTimeoutMilliseconds: 10000
      attributes:
        maxRequestBodySize: 33554432 # 32MB

  platform:
    attributes:
      restartPolicy:
        name: always
        maximumRetryCount: 3
      mountMode: volume