MichalMlodawski
commited on
Commit
•
631537d
1
Parent(s):
7b3a95d
Update config.json
Browse files- config.json +6 -6
config.json
CHANGED
@@ -33,16 +33,16 @@
|
|
33 |
768
|
34 |
],
|
35 |
"id2label": {
|
36 |
-
"0": "
|
37 |
-
"1": "
|
38 |
-
"2": "
|
39 |
},
|
40 |
"image_size": 224,
|
41 |
"initializer_range": 0.02,
|
42 |
"label2id": {
|
43 |
-
"
|
44 |
-
"
|
45 |
-
"
|
46 |
},
|
47 |
"layer_norm_eps": 1e-05,
|
48 |
"layerscale_value": 0.0001,
|
|
|
33 |
768
|
34 |
],
|
35 |
"id2label": {
|
36 |
+
"0": "SAFE",
|
37 |
+
"1": "QUESTIONABLE",
|
38 |
+
"2": "UNSAFE"
|
39 |
},
|
40 |
"image_size": 224,
|
41 |
"initializer_range": 0.02,
|
42 |
"label2id": {
|
43 |
+
"SAFE": 0,
|
44 |
+
"QUESTIONABLE": 1,
|
45 |
+
"UNSAFE": 2
|
46 |
},
|
47 |
"layer_norm_eps": 1e-05,
|
48 |
"layerscale_value": 0.0001,
|