from pathlib import Path
from transformers import LayoutLMv3ForTokenClassification

model_dir = Path('models/extractor_v3') / 'checkpoint-645'
print('Loading model from', model_dir)
model = LayoutLMv3ForTokenClassification.from_pretrained(model_dir)
print('id2label:')
print(model.config.id2label)