mou11 commited on
Commit
1f93cd2
·
verified ·
1 Parent(s): 4f83e03

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": "vit",
3
+ "num_labels": 2,
4
+ "id2label": {
5
+ "0": "NORMAL",
6
+ "1": "PNEUMONIA"
7
+ },
8
+ "label2id": {
9
+ "NORMAL": 0,
10
+ "PNEUMONIA": 1
11
+ },
12
+ "image_size": 224,
13
+ "base_model": "google/vit-base-patch16-224"
14
+ }