update readme

README.md

@@ -12,6 +12,14 @@ Full code and details at https://github.com/csinva/gpt-paper-title-generator
 - inference
     - should prepend with a year and two newlines before querying for a title, e.g. `2022\n\n`
 
+```python    
+from transformers import AutoModelForCausalLM, pipeline, AutoTokenizer
+model = AutoModelForCausalLM.from_pretrained("csinva/gpt-neo-2.7B-titles")
+tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-2.7B")
+pipe = pipeline('text-generation', model=model, tokenizer=tokenizer)
+pipe('2022\n\n')
+```
+
 **Data**
 - all [papers on arXiv](https://www.kaggle.com/datasets/Cornell-University/arxiv) in the categories cs.AI, cs.LG, stat.ML
     - date cutoff: only finetuned on papers with dat on or before Apr 1, 2022