diff --git a/config.json b/config.json
index 421f926..ffe2f5f 100644
--- a/config.json
+++ b/config.json
@@ -41,5 +41,9 @@
   "id2label": {
     "0": "N/A",
     "1": "table",
-  }
+  },
+  "label2id": {
+    "N/A": 0,
+    "table": 1,
+   }
 }