lama9876 commited on
Commit
5bef067
·
verified ·
1 Parent(s): e239160

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +24 -23
config.json CHANGED
@@ -1,23 +1,24 @@
1
- {
2
- "model_type": "image-classification",
3
- "architectures": ["AutoModelForImageClassification"],
4
- "num_labels": 7,
5
- "id2label": {
6
- "0": "Angry",
7
- "1": "Disgust",
8
- "2": "Fear",
9
- "3": "Happy",
10
- "4": "Sad",
11
- "5": "Surprise",
12
- "6": "Neutral"
13
- },
14
- "label2id": {
15
- "Angry": 0,
16
- "Disgust": 1,
17
- "Fear": 2,
18
- "Happy": 3,
19
- "Sad": 4,
20
- "Surprise": 5,
21
- "Neutral": 6
22
- }
23
- }
 
 
1
+ {
2
+ "model_type": "image-classification",
3
+ "architectures": ["Xception"],
4
+ "input_shape": [48, 48, 1],
5
+ "num_labels": 7,
6
+ "id2label": {
7
+ "0": "Angry",
8
+ "1": "Disgust",
9
+ "2": "Fear",
10
+ "3": "Happy",
11
+ "4": "Sad",
12
+ "5": "Surprise",
13
+ "6": "Neutral"
14
+ },
15
+ "label2id": {
16
+ "Angry": 0,
17
+ "Disgust": 1,
18
+ "Fear": 2,
19
+ "Happy": 3,
20
+ "Sad": 4,
21
+ "Surprise": 5,
22
+ "Neutral": 6
23
+ }
24
+ }