exact_data2023/application/functions/errors.py

from transformers import AutoTokenizer
from transformers import pipeline
#from transformers import BartForConditionalGeneration

tokenizer = AutoTokenizer.from_pretrained("szymonj/polish-simple-error-correction")
#model = BartForConditionalGeneration.from_pretrained("szymonj/polish-simple-error-correction")
pipe = pipeline("text2text-generation",model="szymonj/polish-simple-error-correction",tokenizer=tokenizer,max_length=2000)

def errors_correction(data):
    result = pipe(data)
    
    return result