emotionanalysis commited on
Commit
dcc3810
·
verified ·
1 Parent(s): 6868e7a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -8,14 +8,14 @@ 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
 
@@ -36,4 +36,4 @@ outputs = model.generate(input_ids=inputs["input_ids"], max_length=128, num_beam
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
 
 
36
 
37
  generated_comment = tokenizer.decode(outputs[0], skip_special_tokens=True)
38
  print(generated_comment)
39
+ }```