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

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -6
README.md CHANGED
@@ -8,15 +8,13 @@ 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 +33,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
+ ```from transformers import T5TokenizerFast, T5ForConditionalGeneration
 
12
  tokenizer = T5TokenizerFast.from_pretrained("paust/pko-t5-base")
13
+ model = T5ForConditionalGeneration.from_pretrained("emotionanalysis/diaryempathizer-t5-ko")
14
  ```
15
 
16
  Test code
17
+ ```import torch
 
18
  from transformers import T5TokenizerFast, T5ForConditionalGeneration
19
 
20
  model = T5ForConditionalGeneration.from_pretrained("emotionanalysis/diaryempathizer-t5-ko")
 
33
  outputs = model.generate(input_ids=inputs["input_ids"], max_length=128, num_beams=4, early_stopping=True)
34
 
35
  generated_comment = tokenizer.decode(outputs[0], skip_special_tokens=True)
36
+ print(generated_comment)
37
  ```