karthikeya09's picture
Upload config.json with huggingface_hub
ab6812c verified
{
"model_name": "Smart Waste Classification",
"architecture": "MobileNetV2",
"num_classes": 6,
"classes": [
"glass",
"metal",
"non-recyclable",
"organic",
"paper",
"plastic"
],
"image_size": 224,
"framework": "PyTorch",
"pretrained_on": "ImageNet",
"validation_accuracy": "97.46%",
"input_normalization": {
"mean": [
0.485,
0.456,
0.406
],
"std": [
0.229,
0.224,
0.225
]
}
}