diff --git a/config.json b/config.json index adcd723..6974180 100644 --- a/config.json +++ b/config.json @@ -38,4 +38,9 @@ "scale_embedding": false, "torch_dtype": "float32", "transformers_version": "4.17.0" + "id2label": { + "0": "N/A", + "1": "table", + "2": "table", + } }