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