exact_data2023/application/functions/errors.py
Maciej Ścigacz 51d691f08c fix pipelines
2023-05-31 16:56:13 +02:00

9 lines
268 B
Python

from transformers import pipeline
pipe = pipeline("text2text-generation",model="szymonj/polish-simple-error-correction", tokenizer="szymonj/polish-simple-error-correction", max_length=2000)
def errors_correction(data):
result = pipe(data)
return result