andythetechnerd03
/

BERT-tiny_AI-Human

Text Classification

text-embeddings-inference

Model card Files Files and versions

andythetechnerd03 commited on Feb 21, 2024

Commit

c8f46fb

·

verified ·

1 Parent(s): 9e1ec69

Update config.json

Files changed (1) hide show

config.json +6 -1

config.json CHANGED Viewed

@@ -22,5 +22,10 @@
   "transformers_version": "4.37.2",
   "type_vocab_size": 2,
   "use_cache": true,
-  "vocab_size": 30522
 }

   "transformers_version": "4.37.2",
   "type_vocab_size": 2,
   "use_cache": true,
+  "vocab_size": 30522,
+  "label2id" : {"AI": 0,
+                "Human": 1},
+  "id2label" : {0: "Human",
+                1: "AI"}
 }