Create README.md

README.md

@@ -0,0 +1,22 @@
+---
+datasets:
+- IteraTeR_full_sent
+---
+
+# IteraTeR PEGASUS model
+This model was obtained by fine-tuning [google/pegasus-large](https://huggingface.co/google/pegasus-large) on [IteraTeR-full-sent](https://huggingface.co/datasets/wanyu/IteraTeR_full_sent) dataset.
+
+Paper: [Understanding Iterative Revision from Human-Written Text](https://arxiv.org/abs/2203.03802) <br>
+Authors: Wanyu Du, Vipul Raheja, Dhruv Kumar, Zae Myung Kim, Melissa Lopez, Dongyeop Kang
+
+## Usage
+```python
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+
+tokenizer = AutoTokenizer.from_pretrained("wanyu/IteraTeR-PEGASUS")
+model = AutoModelForSeq2SeqLM.from_pretrained("wanyu/IteraTeR-PEGASUS")
+before_input = '<fluency> I likes coffee.'
+model_input = tokenizer(before_input, return_tensors='pt')
+model_outputs = model.generate(**model_input, num_beams=8, max_length=1024)
+after_text = tokenizer.batch_decode(model_outputs, skip_special_tokens=True)[0]
+```