andythetechnerd03
/

BERT-tiny_AI-Human

Text Classification

Inference Endpoints

Model card Files Files and versions Community

andythetechnerd03 commited on Feb 27, 2024

Commit

2bfa6df

·

verified ·

1 Parent(s): b498098

Upload BertForSequenceClassification

Files changed (1) hide show

config.json +9 -1

config.json CHANGED Viewed

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "google/bert_uncased_L-2_H-128_A-2",
   "architectures": [
     "BertForSequenceClassification"
   ],
@@ -8,8 +8,16 @@
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.1,
   "hidden_size": 128,
   "initializer_range": 0.02,
   "intermediate_size": 512,
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,
   "model_type": "bert",

 {
+  "_name_or_path": "andythetechnerd03/BERT-tiny_AI-Human",
   "architectures": [
     "BertForSequenceClassification"
   ],
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.1,
   "hidden_size": 128,
+  "id2label": {
+    "0": "Human",
+    "1": "AI"
+  },
   "initializer_range": 0.02,
   "intermediate_size": 512,
+  "label2id": {
+    "AI": 1,
+    "Human": 0
+  },
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,
   "model_type": "bert",