Update README.md

README.md

@@ -18,12 +18,16 @@ This model is released under the Apache License 2.0.
 
 ## Usage
 Install the required libraries as follows:
-```bash
+```sh
 >>> python -m pip install numpy safetensors sentencepiece torch transformers
+
 ```
 
 Execute the following python code:
 ```python
+tokenizer = AutoTokenizer.from_pretrained("pfnet/plamo-13b-instruct", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("pfnet/plamo-13b-instruct", trust_remote_code=True)
+
 def completion(prompt: str, max_new_tokens: int = 128) -> str:
     inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
     generated_ids = model.generate(
@@ -33,7 +37,8 @@ def completion(prompt: str, max_new_tokens: int = 128) -> str:
         max_new_tokens=max_new_tokens,
         temperature=1,
         top_p=0.95,
-        do_sample=True,)
+        do_sample=True,
+    )
     return tokenizer.decode(generated_ids[0], skip_special_tokens=True, clean_up_tokenization_spaces=True)
 
 def generate_prompt(messages: list) -> str: