Views
No views yet
| Edit Intention | Definition | Example |
|---|---|---|
| clarity | Make the text more formal, concise, readable and understandable. |
Original: It's like a house which anyone can enter in it. Revised: It's like a house which anyone can enter. |
| fluency | Fix grammatical errors in the text. |
Original: In the same year he became the Fellow of the Royal Society. Revised: In the same year, he became the Fellow of the Royal Society. |
| coherence | Make the text more cohesive, logically linked and consistent as a whole. |
Original: Achievements and awards Among his other activities, he founded the Karachi Film Guild and Pakistan Film and TV Academy. Revised: Among his other activities, he founded the Karachi Film Guild and Pakistan Film and TV Academy. |
| style | Convey the writer’s writing preferences, including emotions, tone, voice, etc.. |
Original: She was last seen on 2005-10-22. Revised: She was last seen on October 22, 2005. |
1from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2
3tokenizer = AutoTokenizer.from_pretrained("wanyu/IteraTeR-PEGASUS-Revision-Generator")
4model = AutoModelForSeq2SeqLM.from_pretrained("wanyu/IteraTeR-PEGASUS-Revision-Generator")
5before_input = '<fluency> I likes coffee.'
6model_input = tokenizer(before_input, return_tensors='pt')
7model_outputs = model.generate(**model_input, num_beams=8, max_length=1024)
8after_text = tokenizer.batch_decode(model_outputs, skip_special_tokens=True)[0]