Update README.md

README.md

@@ -12,7 +12,8 @@ It was fine-tuned on several thousand astronomy abstracts collected on Arxiv.
 ## Model Details
 
 ```python
-from transformers import AutoModelForCausalLM
+from transformers import AutoModelForCausalLM, AutoTokenizer
+import torch
 
 online_model = AutoModelForCausalLM.from_pretrained("charlieoneill/falcon-7b-abstracts-2190", torch_dtype=torch.bfloat16,
                                                     device_map="auto", trust_remote_code=True)