Update config.json

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "hustvl/yolos-small",
+  "_name_or_path": "KaraAgroAI/CADI-AI",
   "architectures": [
     "YolosForObjectDetection"
   ],
@@ -15,8 +15,9 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 384,
   "id2label": {
-    "0": "name",
-    "1": "license-plates"
+    "0": "insect",
+    "1": "abiotic",
+    "2": "disease"
   },
   "image_size": [
     512,
@@ -25,8 +26,9 @@
   "initializer_range": 0.02,
   "intermediate_size": 1536,
   "label2id": {
-    "name": 0,
-    "license-plates": 1
+    "insect": 0,
+    "abiotic": 1,
+    "disease": 2,
   },
   "layer_norm_eps": 1e-12,
   "model_type": "yolos",