Update README.md

README.md

@@ -27,9 +27,13 @@ Korean Custom Dataset
 ``` 
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-model_name='davidkim205/komt-mistral-7b-v1'
-model = AutoModelForCausalLM.from_pretrained(model_name)
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained("shleeeee/mistral-7b-wiki")
+tokenizer = AutoTokenizer.from_pretrained("shleeeee/mistral-7b-wiki")
+
+# Use a pipeline as a high-level helper
+from transformers import pipeline
+
+pipe = pipeline("text-generation", model="shleeeee/mistral-7b-wiki")
 ```
 
 ## Evaluation