Update config.json

config.json

@@ -1,9 +1,7 @@
 {
-  "pipeline_tag": "object-detection",
-  "model_type": "yolo",
-  "task": "object_detection",
-  "model_name": "custom_yolo_model",
-  "classes": [
+  "confidence_threshold": 0.25,
+  "iou_threshold": 0.45,
+  "labels": [
     "Title Block",
     "Sheet Title",
     "Sheet Number",
@@ -22,5 +20,24 @@
     "Drawing Sheet",
     "Cover Sheet",
     "Contact Info"
-  ]
-}
+  ],
+  "max_detections": 300,
+  "model_type": "yolov7",
+  "postprocessing": {
+    "apply_nms": true
+  },
+  "preprocessing": {
+    "mean": [
+      0.0,
+      0.0,
+      0.0
+    ],
+    "normalize": true,
+    "std": [
+      1.0,
+      1.0,
+      1.0
+    ]
+  },
+  "transformers_version": "4.47.1"
+}