Update README.md

README.md

@@ -52,48 +52,45 @@ The model has been trained with the following `label2id` and `id2label` mappings
 ### `label2id` Mapping
 
 ```json
-{
-    "I-Amount Including tax": 0,
-    "I-Due Date": 1,
-    "I-Reference Number": 2,
-    "B-Amount Including tax": 3,
-    "I-Customer Name": 4,
-    "O": 5,
-    "I-Issue Date": 6,
-    "B-Amount": 7,
-    "B-Vendor Name": 8,
-    "I-Vendor Name": 9,
-    "B-Customer Name": 10,
-    "B-Due Date": 11,
-    "I-Amount": 12,
-    "B-Reference Number": 13,
-    "B-Issue Date": 14
-}
-
-### `label2id` Mapping
-
-{
-    0: "I-Amount Including tax",
-    1: "I-Due Date",
-    2: "I-Reference Number",
-    3: "B-Amount Including tax",
-    4: "I-Customer Name",
-    5: "O",
-    6: "I-Issue Date",
-    7: "B-Amount",
-    8: "B-Vendor Name",
-    9: "I-Vendor Name",
-    10: "B-Customer Name",
-    11: "B-Due Date",
-    12: "I-Amount",
-    13: "B-Reference Number",
-    14: "B-Issue Date"
-}
+label2id = {
+    'I-Customer Name': 0,
+    'B-Issue Date': 1,
+    'I-Issue Date': 2,
+    'I-Due Date': 3,
+    'I-Amount': 4,
+    'B-Due Date': 5,
+    'O': 6,
+    'B-Amount Including tax': 7,
+    'B-Customer Name': 8,
+    'B-Amount': 9,
+    'I-Amount Including tax': 10,
+    'B-Vendor Name': 11,
+    'I-Vendor Name': 12,
+    'I-Reference Number': 13,
+    'B-Reference Number': 14
+    }
+id2label = {
+    0: 'I-Customer Name',
+    1: 'B-Issue Date',
+    2: 'I-Issue Date',
+    3: 'I-Due Date',
+    4: 'I-Amount',
+    5: 'B-Due Date',
+    6: 'O',
+    7: 'B-Amount Including tax',
+    8: 'B-Customer Name',
+    9: 'B-Amount',
+    10: 'I-Amount Including tax',
+    11: 'B-Vendor Name',
+    12: 'I-Vendor Name',
+    13: 'I-Reference Number',
+    14: 'B-Reference Number'
+    }
 
 
 ## Citation
-@article{Xu2020LayoutLMv2MP,
-  title={LayoutLMv2: Multi-modal Pre-training for Visually-Rich Document Understanding},
+@article{Xu2020LayoutLM,
+  title={LayoutLM: Multi-modal Pre-training for Visually-Rich Document Understanding},
   author={Yiheng Xu and Minghao Li and Lei Cui and Shaohan Huang and Furu Wei and Ming Zhou},
   journal={ArXiv},
   year={2020},