config.json

{
  "_name_or_path": "alfiannajih/trash-classification",
  "architectures": [
    "MobileNetV3Model"
  ],
  "auto_map": {
    "AutoConfig": "alfiannajih/trash-classification--config.MobileNetV3Config",
    "AutoModelForImageClassification": "alfiannajih/trash-classification--model.MobileNetV3Model"
  },
  "custom_pipelines": {
    "image-classification": {
      "impl": "pipeline.TrashClassificationPipeline",
      "pt": [
        "AutoModelForImageClassification"
      ],
      "tf": [],
      "type": "image"
    }
  },
  "id2label": {
    "0": "cardboard",
    "1": "glass",
    "2": "metal",
    "3": "paper",
    "4": "plastic",
    "5": "trash"
  },
  "model_type": "mobilenetv3",
  "num_classes": 6,
  "torch_dtype": "float32",
  "transformers_version": "4.46.3"
}