Update README.md

README.md

@@ -176,14 +176,19 @@ pip install flash-attn
 ```
 Then load the model in transformers:
 ```python
-from transformers import AutoModelForCausalLM, AutoTokenizer
-import torch
+from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
 
-model = AutoModelForCausalLM.from_pretrained(
-    "MediaTek-Research/Breeze-7B-Instruct-v0.1",
-    device_map="auto",
-    torch_dtype=torch.bfloat16,
+model = AutoModelForCausalLM.from_pretrained("MediaTek-Research/Breeze-7B-Instruct-v0.1")
+tokenizer = AutoTokenizer.from_pretrained("MediaTek-Research/Breeze-7B-Instruct-v0.1")
+
+# you can also using pipeline
+generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
+generator(
+    "請問台灣最高的山是",
+    max_length=30,
+    num_return_sequences=1,
 )
+
 ```
 
 The structure of the query template follows that of Mistral-7B-Instruct, as shown below.