update model trained 3 epochs

config.json

@@ -1,4 +1,5 @@
 {
+  "_name_or_path": "distilbert-base-uncased",
   "activation": "gelu",
   "architectures": [
     "DistilBertForSequenceClassification"
@@ -6,16 +7,15 @@
   "attention_dropout": 0.1,
   "dim": 768,
   "dropout": 0.1,
-  "dtype": "float32",
   "hidden_dim": 3072,
   "id2label": {
-    "0": "Chat",
-    "1": "Search"
+    "0": "Search",
+    "1": "Chat"
   },
   "initializer_range": 0.02,
   "label2id": {
-    "Chat": 0,
-    "Search": 1
+    "Chat": 1,
+    "Search": 0
   },
   "max_position_embeddings": 512,
   "model_type": "distilbert",
@@ -27,6 +27,7 @@
   "seq_classif_dropout": 0.2,
   "sinusoidal_pos_embds": false,
   "tie_weights_": true,
-  "transformers_version": "4.56.1",
+  "torch_dtype": "float32",
+  "transformers_version": "4.49.0",
   "vocab_size": 30522
 }

label_map.json

@@ -1,10 +1,10 @@
 {
   "label2id": {
-    "Chat": 0,
-    "Search": 1
+    "Chat": 1,
+    "Search": 0
   },
   "id2label": {
-    "0": "Chat",
-    "1": "Search"
+    "1": "Chat",
+    "0": "Search"
   }
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9adeb1626e0be3330486338ab2ef906421c3858e10661c9cbefd5a511df0fab6
+oid sha256:b647847cf2dfba5b7698042356aea3f91b44c9031c1787b0eeee682735cffa5a
 size 267832560

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cf15efeda516f29d2cb4fdfbb1c5d25a7d1280056febf0b01eb371cb9c34a990
-size 5777
+oid sha256:0fb0d1b6fd3823850dc4e8ee2d1f9809b2eb7dee96352f075b4459b2e6868b4a
+size 5713