Create config.json
Browse files- config.json +14 -0
config.json
ADDED
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"model_type": "keras",
|
3 |
+
"num_labels": 2, // Adjust this based on your model's output
|
4 |
+
"label2id": {
|
5 |
+
"no_tumor": 0,
|
6 |
+
"tumor": 1
|
7 |
+
},
|
8 |
+
"id2label": {
|
9 |
+
"0": "no_tumor",
|
10 |
+
"1": "tumor"
|
11 |
+
},
|
12 |
+
"image_size": [224, 224], // Adjust this based on your model's input size
|
13 |
+
"framework": "tf" // Specify the framework if you're using TensorFlow/Keras
|
14 |
+
}
|