taeminlee commited on
Commit
de371e7
β€’
1 Parent(s): e60f19e

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +24 -0
README.md ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # KoGPT2-Transformers
2
+
3
+ KoGPT2 on Huggingface Transformers
4
+
5
+ ### KoGPT2-Transformers
6
+
7
+ - [SKT-AI μ—μ„œ κ³΅κ°œν•œ KoGPT2](https://github.com/SKT-AI/KoGPT2)λ₯Ό [Transformers](https://github.com/huggingface/transformers)μ—μ„œ μ‚¬μš©ν•˜λ„λ‘ ν•˜μ˜€μŠ΅λ‹ˆλ‹€.
8
+
9
+ ### Demo
10
+
11
+ - 일상 λŒ€ν™” 챗봇 : http://demo.tmkor.com:36200/dialo
12
+ - ν™”μž₯ν’ˆ 리뷰 생성 : http://demo.tmkor.com:36200/ctrl
13
+
14
+ ### Example
15
+
16
+ ```
17
+ from transformers import GPT2LMHeadModel, PreTrainedTokenizerFast
18
+
19
+ input_ids = tokenizer.encode("μ•ˆλ…•", add_special_tokens=False, return_tensors="pt")
20
+ output_sequences = model.generate(input_ids=input_ids, do_sample=True, max_length=100, num_return_sequences=3)
21
+ for generated_sequence in output_sequences:
22
+ generated_sequence = generated_sequence.tolist()
23
+ print("GENERATED SEQUENCE : {0}".format(tokenizer.decode(generated_sequence, clean_up_tokenization_spaces=True)))
24
+ ```