duongna commited on
Commit
39f06cd
1 Parent(s): 57c57f1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -14,6 +14,7 @@ Details will be available soon.
14
  For more information, please contact anhduongng.1001@gmail.com (Dương) / imthanhlv@gmail.com (Thành) / nguyenvulebinh@gmail.com (Bình).
15
 
16
  ### How to use
 
17
  from transformers import AutoTokenizer, AutoModelForCausalLM
18
 
19
  tokenizer = AutoTokenizer.from_pretrained("VietAI/gpt-j-6B-vietnamese-news")
@@ -34,4 +35,5 @@ gen_tokens = model.generate(
34
  )
35
 
36
  gen_text = tokenizer.batch_decode(gen_tokens)[0]
37
- print(gen_text)
 
14
  For more information, please contact anhduongng.1001@gmail.com (Dương) / imthanhlv@gmail.com (Thành) / nguyenvulebinh@gmail.com (Bình).
15
 
16
  ### How to use
17
+ ```python
18
  from transformers import AutoTokenizer, AutoModelForCausalLM
19
 
20
  tokenizer = AutoTokenizer.from_pretrained("VietAI/gpt-j-6B-vietnamese-news")
35
  )
36
 
37
  gen_text = tokenizer.batch_decode(gen_tokens)[0]
38
+ print(gen_text)
39
+ ```