rubysmac commited on
Commit
26b955e
·
verified ·
1 Parent(s): c37fb47

Third upload of the fine-tuned model

Browse files
Files changed (1) hide show
  1. config.json +23 -23
config.json CHANGED
@@ -1,34 +1,34 @@
1
  {
2
- "_name_or_path": "prithivMLmods/Augmented-Waste-Classifier-SigLIP2",
3
  "architectures": [
4
  "SiglipForImageClassification"
5
  ],
6
  "id2label": {
7
- "0": "LABEL_0",
8
- "1": "LABEL_1",
9
- "2": "LABEL_2",
10
- "3": "LABEL_3",
11
- "4": "LABEL_4",
12
- "5": "LABEL_5",
13
- "6": "LABEL_6",
14
- "7": "LABEL_7",
15
- "8": "LABEL_8",
16
- "9": "LABEL_9",
17
- "10": "LABEL_10"
18
  },
19
  "initializer_factor": 1.0,
20
  "label2id": {
21
- "LABEL_0": 0,
22
- "LABEL_1": 1,
23
- "LABEL_10": 10,
24
- "LABEL_2": 2,
25
- "LABEL_3": 3,
26
- "LABEL_4": 4,
27
- "LABEL_5": 5,
28
- "LABEL_6": 6,
29
- "LABEL_7": 7,
30
- "LABEL_8": 8,
31
- "LABEL_9": 9
32
  },
33
  "model_type": "siglip",
34
  "problem_type": "single_label_classification",
 
1
  {
2
+ "_name_or_path": "rubysmac/compostable-plates-classifier-v0",
3
  "architectures": [
4
  "SiglipForImageClassification"
5
  ],
6
  "id2label": {
7
+ "0": "Battery",
8
+ "1": "Biological",
9
+ "2": "Cardboard",
10
+ "3": "Clothes",
11
+ "4": "Glass",
12
+ "5": "Metal",
13
+ "6": "Paper",
14
+ "7": "Plastic",
15
+ "8": "Shoes",
16
+ "9": "Trash",
17
+ "10": "Compostable"
18
  },
19
  "initializer_factor": 1.0,
20
  "label2id": {
21
+ "Battery": 0,
22
+ "Biological": 1,
23
+ "Cardboard": 2,
24
+ "Clothes": 3,
25
+ "Compostable": 10,
26
+ "Glass": 4,
27
+ "Metal": 5,
28
+ "Paper": 6,
29
+ "Plastic": 7,
30
+ "Shoes": 8,
31
+ "Trash": 9
32
  },
33
  "model_type": "siglip",
34
  "problem_type": "single_label_classification",