Update README.md

README.md

@@ -42,6 +42,9 @@ Install the required libraries as follows:
 
 Execute the following python code:
 ```python
+import torch
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
 tokenizer = AutoTokenizer.from_pretrained("pfnet/nekomata-14b-pfn-qfin", trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained("pfnet/nekomata-14b-pfn-qfin", device_map="auto", trust_remote_code=True)
 text = "日本銀行は"