Update config.json

config.json

@@ -1,10 +1,8 @@
 {
-  "task": "detect",
-  "mode": "predict",
-  "conf": 0.5,
-  "imgsz": 640,
-  "nc": 17,
-  "model_type": "yolos",
+  "_name_or_path": "RaphaelKalandadze/tmp_trainer",
+  "architectures": [
+    "YolosForObjectDetection"
+  ],
   "id2label": {
     "0":"aop", 
     "1":"asc", 
@@ -24,6 +22,10 @@
     "15":"vegetarian", 
     "16":"vpf"
   },
+  "image_size": [
+    640,
+    640
+  ],
   "label2id": {
     "aop":0, 
     "asc":1, 
@@ -42,5 +44,8 @@
     "vbf":14, 
     "vegetarian":15, 
     "vpf":16
-  }
+  },
+  "model_type": "yolos",
+  "num_channels": 3,
+  "transformers_version": "4.21.1",
 }