Initial release of InferenceGuard RoBERTa risk classifier

config.json

@@ -1,31 +1,6 @@
 {
-  "transformers_version": "5.22.1",
-  "architectures": null,
-  "output_hidden_states": false,
-  "return_dict": true,
-  "dtype": null,
-  "chunk_size_feed_forward": 0,
-  "is_encoder_decoder": false,
-  "id2label": {
-    "0": "LABEL_0",
-    "1": "LABEL_1"
-  },
-  "label2id": {
-    "LABEL_0": 0,
-    "LABEL_1": 1
-  },
-  "problem_type": null,
-  "_name_or_path": "",
-  "base_model_name": "roberta-base",
-  "num_heads": 4,
-  "num_classes_per_head": 1,
-  "hidden_size": 768,
-  "classifier_dropout": 0.1,
-  "temperature_values": [
-    1.0,
-    1.0,
-    1.0,
-    1.0
+  "architectures": [
+    "RobertaRiskClassifier"
   ],
   "attributes": [
     "age",
@@ -33,6 +8,18 @@
     "occupation",
     "education"
   ],
+  "base_model_name": "roberta-base",
+  "classifier_dropout": 0.1,
+  "dtype": "float32",
+  "hidden_size": 768,
   "model_type": "inferenceguard-roberta",
-  "output_attentions": false
-}
+  "num_classes_per_head": 1,
+  "num_heads": 4,
+  "temperature_values": [
+    1.0,
+    1.0,
+    1.0,
+    1.0
+  ],
+  "transformers_version": "5.20.2"
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:40c4073349ff3827f09503349ec1be518a66bc234991af7fb39c46f3bf8255fd
+size 503339184