emotionanalysis commited on
Commit
0e4567d
·
verified ·
1 Parent(s): dcc3810

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -8
README.md CHANGED
@@ -8,15 +8,15 @@ pipeline_tag: text2text-generation
8
  paust/pko-t5-base model based
9
 
10
  Since this model is based on paust/pko-t5-base tokenizer, you need to import it.
11
- ```{
12
- from transformers import T5TokenizerFast, T5ForConditionalGeneration
13
  tokenizer = T5TokenizerFast.from_pretrained("paust/pko-t5-base")
14
- model = T5ForConditionalGeneration.from_pretrained("emotionanalysis/diaryempathizer-t5-ko")
15
- }```
16
 
17
  Test code
18
- ```{
19
- import torch
20
  from transformers import T5TokenizerFast, T5ForConditionalGeneration
21
 
22
  model = T5ForConditionalGeneration.from_pretrained("emotionanalysis/diaryempathizer-t5-ko")
@@ -35,5 +35,5 @@ inputs = {key: value.to(device) for key, value in inputs.items()}
35
  outputs = model.generate(input_ids=inputs["input_ids"], max_length=128, num_beams=4, early_stopping=True)
36
 
37
  generated_comment = tokenizer.decode(outputs[0], skip_special_tokens=True)
38
- print(generated_comment)
39
- }```
 
8
  paust/pko-t5-base model based
9
 
10
  Since this model is based on paust/pko-t5-base tokenizer, you need to import it.
11
+ ```
12
+ {from transformers import T5TokenizerFast, T5ForConditionalGeneration
13
  tokenizer = T5TokenizerFast.from_pretrained("paust/pko-t5-base")
14
+ model = T5ForConditionalGeneration.from_pretrained("emotionanalysis/diaryempathizer-t5-ko")}
15
+ ```
16
 
17
  Test code
18
+ ```
19
+ {import torch
20
  from transformers import T5TokenizerFast, T5ForConditionalGeneration
21
 
22
  model = T5ForConditionalGeneration.from_pretrained("emotionanalysis/diaryempathizer-t5-ko")
 
35
  outputs = model.generate(input_ids=inputs["input_ids"], max_length=128, num_beams=4, early_stopping=True)
36
 
37
  generated_comment = tokenizer.decode(outputs[0], skip_special_tokens=True)
38
+ print(generated_comment)}
39
+ ```