beomi commited on
Commit
92f2c3e
β€’
1 Parent(s): f000713

update usecase with pipeline on README

Browse files
Files changed (1) hide show
  1. README.md +5 -5
README.md CHANGED
@@ -14,9 +14,9 @@ language: ko
14
  * Check the model performance and other language models for Korean in [github](https://github.com/kiyoungkim1/LM-kor)
15
 
16
  ```python
17
- from transformers import BertTokenizerFast, GPT2LMHeadModel
18
- tokenizer_gpt3 = BertTokenizerFast.from_pretrained("kykim/gpt3-kor-small_based_on_gpt2")
19
- input_ids = tokenizer_gpt3.encode("text to tokenize")[1:] # remove cls token
20
-
21
- model_gpt3 = GPT2LMHeadModel.from_pretrained("kykim/gpt3-kor-small_based_on_gpt2")
22
  ```
 
14
  * Check the model performance and other language models for Korean in [github](https://github.com/kiyoungkim1/LM-kor)
15
 
16
  ```python
17
+ from transformers import pipeline
18
+
19
+ pipe = pipeline('text-generation', model='beomi/kykim-gpt3-kor-small_based_on_gpt2')
20
+ print(pipe("μ•ˆλ…•ν•˜μ„Έμš”! μ˜€λŠ˜μ€"))
21
+ # [{'generated_text': 'μ•ˆλ…•ν•˜μ„Έμš”! μ˜€λŠ˜μ€ μ œκ°€ μš”μ¦˜ μ‚¬μš©ν•˜κ³  μžˆλŠ” ν΄λ Œμ§•μ›Œν„°λ₯Ό μ†Œκ°œν•΄λ“œλ¦¬λ €κ³  ν•΄μš”! λ°”λ‘œ 이 μ œν’ˆ!! λ°”λ‘œ 이'}]
22
  ```