Update README.md

README.md

@@ -2,21 +2,22 @@ data: https://github.com/BigSalmon2/InformalToFormalDataset
 
 Text Generation Informal Formal
 
+Trained on this model: https://huggingface.co/CarperAI/FIM-NeoX-1.3B, which is geared toward filling in the blank.
+
 ```
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln82Paraphrase")
-
 model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln82Paraphrase")
 ```
 
 ```
 Demo:
-https://huggingface.co/spaces/BigSalmon/FormalInformalConciseWordy
+https://huggingface.co/spaces/BigSalmon/GPT2Mask
 ```
 
 ```
-prompt = """informal english: corn fields are all across illinois, visible once you leave chicago.\nTranslated into the Style of Abraham Lincoln:"""
+prompt = """<|SUF|> into relaxation <|PRE|> music before bedtime <|MID|>"""
 input_ids = tokenizer.encode(prompt, return_tensors='pt')
 outputs = model.generate(input_ids=input_ids,
                              max_length=10 + len(prompt),