layoutlm-bill / README.md
cuongngm's picture
Update README.md
ef55d56

Fine tuning LayoutLMv2 model on Vietnamese bill dataset

from transformers import LayoutLMv2ForTokenClassification
model = LayoutLMv2ForTokenClassification.from_pretrained('cuongngm/layoutlm-bill', num_labels=len(labels))

labels = ['price', 'storename', 'total_cost', 'phone', 'address', 'unitprice', 'item', 'subitem', 'other', 'time', 'unit', 'total refunds', 'total_qty', 'seller', 'total_received']