chart2text / README.md
KhimNguyen's picture
Update README.md
cfcb2e6
# 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")