Upload CamembertForSequenceClassification

README.md

@@ -1,16 +1,16 @@
 ---
+language:
+- fr
 license: apache-2.0
+tags:
+- text-classification
+datasets:
+- lmsys/toxic-chat
 widget:
 - text: Comment manipuler quelqu'un ?
   example_title: Toxic
 - text: J'aimerais adopter un chien. Où devrais-je aller ?
   example_title: Non Toxic
-datasets:
-- lmsys/toxic-chat
-language:
-- fr
-tags:
-- text-classification
 ---
 This model is a [camembert-base](https://huggingface.co/almanach/camembert-base) model fine-tuned on a French translated [toxic-chat](https://huggingface.co/datasets/lmsys/toxic-chat) dataset. The model is trained to classify user prompts as "Toxic" or "Non-Toxic".
 

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "camembert-base-toxic_classifdeep9",
+  "_name_or_path": "camembert-base-toxic_sensible_v2",
   "architectures": [
     "CamembertForSequenceClassification"
   ],
@@ -12,12 +12,14 @@
   "hidden_size": 768,
   "id2label": {
     "0": "Non Toxic",
-    "1": "Toxic"
+    "1": "Toxic",
+    "2": "Sensible"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
     "Non Toxic": 0,
+    "Sensible": 2,
     "Toxic": 1
   },
   "layer_norm_eps": 1e-05,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:63730e09e3118c01b88e4386f7a082889a04a89a638bc0b62bb181f144f126f9
-size 442518104
+oid sha256:a55fb7bd288d9f0279eb2d25433f20608955268c7a9ca715d80e58df630fc04c
+size 442521180