carbon225
/

vit-base-patch16-224-hentai

Image Classification

nsfw

Inference Endpoints

Model card Files Files and versions Community

Carbon225 commited on Sep 30, 2022

Commit

c810a47

·

1 Parent(s): fb38a40

fix config

Files changed (1) hide show

config.json +5 -5

config.json CHANGED Viewed

@@ -8,11 +8,11 @@
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
-  "id2label": [
-    "safe",
-    "suggestive",
-    "explicit"
-  ],
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,

   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
+  "id2label": {
+    "0": "safe",
+    "1": "suggestive",
+    "2": "explicit"
+  },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,