{ | |
"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 | |
} | |