michellejieli
/

NSFW_text_classifier

Text Classification

NSFW

inappropriate

Inference Endpoints

Model card Files Files and versions Community

michellejieli commited on Dec 10, 2022

Commit

b875214

•

1 Parent(s): 327f5ef

Change label names

Files changed (1) hide show

config.json +5 -5

config.json CHANGED Viewed

@@ -9,13 +9,13 @@
   "dropout": 0.1,
   "hidden_dim": 3072,
   "id2label": {
-    "0": "NSFW",
-    "1": "SFW"
   },
   "initializer_range": 0.02,
   "label2id": {
-    "NSFW": 0,
-    "SFW": 1
   },
   "max_position_embeddings": 512,
   "model_type": "distilbert",
@@ -30,4 +30,4 @@
   "torch_dtype": "float32",
   "transformers_version": "4.22.2",
   "vocab_size": 30522
-}

   "dropout": 0.1,
   "hidden_dim": 3072,
   "id2label": {
+    "0": "SFW",
+    "1": "NSFW"
   },
   "initializer_range": 0.02,
   "label2id": {
+    "SFW": 0,
+    "NSFW": 1
   },
   "max_position_embeddings": 512,
   "model_type": "distilbert",
   "torch_dtype": "float32",
   "transformers_version": "4.22.2",
   "vocab_size": 30522
+}