| 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 | |