pavankm96 commited on
Commit
864e141
·
verified ·
1 Parent(s): ee26137

Create config.json

Browse files
Files changed (1) hide show
  1. 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
+ }