shevek commited on
Commit
7587ba5
1 Parent(s): be3b6eb

Create config.yaml

Browse files
Files changed (1) hide show
  1. config.yaml +19 -0
config.yaml ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model_type: object-detection
2
+ model:
3
+ name: Doors_YOLOv8
4
+ path: yolov8n.pt
5
+
6
+ labels:
7
+ - door
8
+
9
+ inference:
10
+ tokenizer:
11
+ class: transformers.PreTrainedTokenizerFast
12
+ pretrained: 'ultralytics/yolov8'
13
+ model:
14
+ class: torch.hub.load
15
+ path: 'Madronus/Doors_YOLOv8:master', 'yolov8n.pt' # Assuming using the Ultralytics' hub reference for YOLOv8
16
+ force_download: true
17
+
18
+ pip_dependencies:
19
+ # - yolov8 # Assuming there's a pip package named 'yolov8', you might need to change this to the correct dependency