Create config.json

config.json

@@ -0,0 +1,14 @@
+{
+  "model_type": "resnet",
+  "num_labels": 2,
+  "labels": {
+    "0": "diagram",
+    "1": "none"
+  },
+  "input_size": [224, 224],
+  "normalization": {
+    "mean": [0.485, 0.456, 0.406],
+    "std": [0.229, 0.224, 0.225]
+  },
+  "framework": "pytorch"
+}