Update README.md

README.md

@@ -63,7 +63,7 @@ print(gen_text)
 ## Usage with pipeline
 
 ```python
-from transformers import pipeline
+from transformers import pipeline, GPTNeoXForCausalLM, GPTNeoXTokenizerFast
 
 model = GPTNeoXForCausalLM.from_pretrained("NYTK/PULI-GPT-3SX")
 tokenizer = GPTNeoXTokenizerFast.from_pretrained("NYTK/PULI-GPT-3SX")