Update README.md

README.md

@@ -25,10 +25,11 @@ Korean Custom Dataset
 
 ## Usage
 ``` 
+# Load model directly
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-model = AutoModelForCausalLM.from_pretrained("shleeeee/mistral-7b-wiki")
 tokenizer = AutoTokenizer.from_pretrained("shleeeee/mistral-7b-wiki")
+model = AutoModelForCausalLM.from_pretrained("shleeeee/mistral-7b-wiki")
 
 # Use a pipeline as a high-level helper
 from transformers import pipeline