Update README.md
Browse files
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 |
+
}```
|