simondg
2nd version + model cards addition
af74633
|
raw
history blame
No virus
888 Bytes

ByT5 Dutch OCR correction

This model is a finetuned byT5 model that corrects OCR mistakes found in dutch sentences. The google/byt5-base model is finetuned on the dutch section of the OSCAR dataset.

Usage

from transformers import AutoTokenizer, T5ForConditionalGeneration

example_sentence = "Een algoritme dat op basis van kunstmatige inte11i9entie vkijwe1 geautomatiseerd een Nederlandstalige tekst samenstelt."

tokenizer = AutoTokenizer.from_pretrained('ml6team/byt5-small-dutch-ocr-correction')

model_inputs = tokenizer(example_sentence, max_length=128, truncation=True, return_tensors="pt")

model = T5ForConditionalGeneration.from_pretrained('ml6team/byt5-small-dutch-ocr-correction')
outputs = model.generate(**model_inputs, max_length=128)

tokenizer.decode(outputs[0])