model_type: object-detection
model:
  name: Doors_YOLOv8
  path: yolov8n.pt

labels:
  - door

inference:
  tokenizer:
    class: transformers.PreTrainedTokenizerFast
    pretrained: 'ultralytics/yolov8'
  model:
    class: torch.hub.load
    path: 'Madronus/Doors_YOLOv8:master', 'yolov8n.pt' # Assuming using the Ultralytics' hub reference for YOLOv8
    force_download: true

pip_dependencies:
  # - yolov8 # Assuming there's a pip package named 'yolov8', you might need to change this to the correct dependency