# Load model directly
from transformers import AutoProcessor, AutoModelForTokenClassification
processor = AutoProcessor.from_pretrained("cuongngm/layoutlm-bill")
model = AutoModelForTokenClassification.from_pretrained("cuongngm/layoutlm-bill")Quick Links
YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
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']
- Downloads last month
- 8
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="cuongngm/layoutlm-bill")