from transformers import pipeline

model="/data/scratch/svanhvit/byt5/byt5-base-gec-and-ged-iec-ckpt-550K-is_err-to-is_corr-max-len-512-bz-4-lr-1e-05/checkpoint-70000"

gec_pipeline = pipeline(
    'text2text-generation',
    model=model,
    tokenizer="google/byt5-base"
    )

while True:
    sentence = input("Skrifaðu setningu til að flokka: " + "\n")
    print(gec_pipeline("has_error " + sentence.strip(), max_length=512)[0]["generated_text"] + "\n")