brain_tumor_det / config.json
pavankm96's picture
Create config.json
864e141 verified
raw
history blame
395 Bytes
{
"model_type": "keras",
"num_labels": 2, // Adjust this based on your model's output
"label2id": {
"no_tumor": 0,
"tumor": 1
},
"id2label": {
"0": "no_tumor",
"1": "tumor"
},
"image_size": [224, 224], // Adjust this based on your model's input size
"framework": "tf" // Specify the framework if you're using TensorFlow/Keras
}