keffy commited on
Commit
7d2473c
1 Parent(s): 0826502

Update model-index.yaml

Browse files
Files changed (1) hide show
  1. model-index.yaml +3 -6
model-index.yaml CHANGED
@@ -1,13 +1,10 @@
1
  model-index:
2
- - name: Rock vs Non-Rock Model
3
  task:
4
  type: object-detection
5
- framework: pytorch
6
- tags:
7
- - yolov8
8
- - rock-classification
9
- config: config.yaml
10
  weights: best.pt
 
11
  labels:
12
  0: rock
13
  1: non-rock
 
1
  model-index:
2
+ - name: Rock vs Non-Rock YOLOv8 Model
3
  task:
4
  type: object-detection
5
+ framework: ultralytics
 
 
 
 
6
  weights: best.pt
7
+ config: config.yaml
8
  labels:
9
  0: rock
10
  1: non-rock