update readme

README.md

@@ -3,7 +3,7 @@ language: "en"
 tags:
 - counterfactual generation
 widget:
-- text: "It is great for kids. <perturb> [negation] It [BLANK] great for kids. [SEP]"
+- text: "A dog is embraced by the woman. <perturb> [negation] A dog is [BLANK] the woman."
 ---
 
 # Polyjuice
@@ -17,11 +17,12 @@ This is a ported version of [Polyjuice](https://homes.cs.washington.edu/~wtshuan
 ```python
 from transformers import AutoTokenizer, AutoModelWithLMHead
 
-tokenizer = AutoTokenizer.from_pretrained("uw-hai/jolyjuice")
-model = AutoModelWithLMHead.from_pretrained("uw-hai/jolyjuice")
+tokenizer = AutoTokenizer.from_pretrained("uw-hai/polyjuice")
+model = AutoModelWithLMHead.from_pretrained("uw-hai/polyjuice")
 
 
-prompt_text = "It is great for kids. <perturb> [negation] It [BLANK] great for kids. [SEP]"
+prompt_text = "A dog is embraced by the woman. <perturb> [negation] A dog is [BLANK] the woman."
+# or try: "A dog is embraced by the woman. <perturb> [restructure] A dog is [BLANK] the woman."
 perturb_tok, end_tok = "<|perturb|>", "<|endoftext|>"
 encoded_prompt = tokenizer.encode(prompt_text, add_special_tokens=False, return_tensors="pt")
 input_ids = encoded_prompt