File size: 661 Bytes
cfcb2e6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
# Fine-tune Donut to extract data from chart
## Data
The data used to train, validate and test is published by account named TeeA via this link huggingface.co/datasets/TeeA/Vietnamese-Chart-Dataset
## Fine-tuning instruction
The model is fine-tuned following the instruction of Niels Rogge - Transformer Tutorial, 2020-09-02 via https://github.com/NielsRogge/Transformers-Tutorials
## Load model
The model can be loaded by using DonutProcessor
from transformers import DonutProcessor, VisionEncoderDecoderModel
processor = DonutProcessor.from_pretrained("KhimNguyen/chart2text")
model = VisionEncoderDecoderModel.from_pretrained("KhimNguyen/chart2text") |