Hugging Face's logo

alfiannajih
/
trash-classification

Image Classification
Transformers
Safetensors
mobilenetv3
custom_code
Model card Files Community
alfiannajih commited on
Commit
3283c80
·
verified ·
1 Parent(s): 9832ccc

Upload config

Browse files
Files changed (2) hide show
  1. config.json +9 -16
  2. config.py +2 -2
config.json CHANGED
@@ -1,24 +1,17 @@
1
  {
2
- "_name_or_path": "alfiannajih/trash-classification",
3
- "architectures": [
4
- "MobileNetV3Model"
5
- ],
6
  "auto_map": {
7
- "AutoConfig": "alfiannajih/trash-classification--config.MobileNetV3Config",
8
- "AutoModelForImageClassification": "alfiannajih/trash-classification--model.MobileNetV3Model"
9
  },
10
- "custom_pipelines": {
11
- "image-classification": {
12
- "impl": "pipeline.TrashClassificationPipeline",
13
- "pt": [
14
- "AutoModelForImageClassification"
15
- ],
16
- "tf": [],
17
- "type": "image"
18
- }
19
  },
20
  "model_type": "mobilenetv3",
21
  "num_classes": 6,
22
- "torch_dtype": "float32",
23
  "transformers_version": "4.46.3"
24
  }
 
1
  {
2
+ "_attn_implementation_autoset": true,
 
 
 
3
  "auto_map": {
4
+ "AutoConfig": "config.MobileNetV3Config"
 
5
  },
6
+ "id2label": {
7
+ "0": "cardboard",
8
+ "1": "glass",
9
+ "2": "metal",
10
+ "3": "paper",
11
+ "4": "plastic",
12
+ "5": "trash"
 
 
13
  },
14
  "model_type": "mobilenetv3",
15
  "num_classes": 6,
 
16
  "transformers_version": "4.46.3"
17
  }
config.py CHANGED
@@ -17,6 +17,6 @@ class MobileNetV3Config(PretrainedConfig):
17
  },
18
  **kwargs,
19
  ):
20
- self.num_classes = num_classes
21
  self.id2label = id2label
22
- super().__init__(**kwargs)
 
17
  },
18
  **kwargs,
19
  ):
20
+ super().__init__(**kwargs)
21
  self.id2label = id2label
22
+ self.num_classes = num_classes