Update config.json

config.json

@@ -1,20 +1,35 @@
 {
+  "model_type": "densenet",
+  "architectures": [
+    "DenseNetForImageClassification"
+  ],
+  "pipeline_tag": "image-classification",
+  "transformers_version": "4.30.0",
+  "image_size": [
+    224,
+    224
+  ],
+  "normalize": {
+    "mean": [
+      0.485,
+      0.456,
+      0.406
+    ],
+    "std": [
+      0.229,
+      0.224,
+      0.225
+    ]
+  },
   "weights_url": "https://github.com/mlmed/torchxrayvision/releases/download/v1/chex-densenet121-d121-tw-lr001-rot45-tr15-sc15-seed0-best.pt",
   "labels": [
     "Atelectasis",
     "Consolidation",
-    "",
     "Pneumothorax",
     "Edema",
-    "",
-    "",
     "Effusion",
     "Pneumonia",
-    "",
     "Cardiomegaly",
-    "",
-    "",
-    "",
     "Lung Lesion",
     "Fracture",
     "Lung Opacity",
@@ -23,21 +38,14 @@
   "op_threshs": [
     0.1988969,
     0.05710573,
-    NaN,
     0.0531293,
     0.1435217,
-    NaN,
-    NaN,
     0.27212676,
     0.07749717,
-    NaN,
     0.19712369,
-    NaN,
-    NaN,
-    NaN,
     0.09932402,
     0.09273402,
     0.3270967,
     0.10888247
   ]
-}
+}