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: "It is great for kids. <|perturb|> [negation] It [BLANK] great for kids. [SEP]"
 ---
 
 # Polyjuice
@@ -21,8 +21,8 @@ tokenizer = AutoTokenizer.from_pretrained("uw-hai/polyjuice")
 model = AutoModelWithLMHead.from_pretrained("uw-hai/polyjuice")
 
 
-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."
+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