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