pneumonia-detector / config.json
mou11's picture
Create config.json
1f93cd2 verified
{
"model_type": "vit",
"num_labels": 2,
"id2label": {
"0": "NORMAL",
"1": "PNEUMONIA"
},
"label2id": {
"NORMAL": 0,
"PNEUMONIA": 1
},
"image_size": 224,
"base_model": "google/vit-base-patch16-224"
}