Update README.md

README.md

@@ -31,4 +31,8 @@ prompt = "define \"noseplow\": The children hid as the noseplow drove across the
 ids = tokenizer(prompt, return_tensors="pt").input_ids
 generated_tokens = model.generate(ids)[0][1:-1]
 print(tokenizer.decode(generated_tokens))
-```
+```
+
+See the gist for the source code to used to train the model:
+
+https://gist.github.com/marksverdhei/0a13f67e65460b71c05fcf558a6a91ae