wanyu
/

IteraTeR-PEGASUS-Revision-Generator

text2text-generation

Model card Files Files and versions

wanyu commited on Mar 20, 2022

Commit

c684f1c

·

1 Parent(s): f741f23

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -56,8 +56,8 @@ The edit intentions are provided by [IteraTeR-full-sent](https://huggingface.co/
 ```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)

 ```python
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+tokenizer = AutoTokenizer.from_pretrained("wanyu/IteraTeR-PEGASUS-Revision-Generator")
+model = AutoModelForSeq2SeqLM.from_pretrained("wanyu/IteraTeR-PEGASUS-Revision-Generator")
 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)