Update README.md
Browse files
README.md
CHANGED
@@ -42,11 +42,11 @@ out = model.generate(tokenized_poet_start,
|
|
42 |
num_beams=8,
|
43 |
no_repeat_ngram_size=2,
|
44 |
early_stopping=True,
|
45 |
-
pad_token_id= tokenizer.pad_token_id
|
46 |
eos_token_id = tokenizer.eos_token_id)
|
47 |
|
48 |
# Decode Poet
|
49 |
-
decoded_cont =
|
50 |
|
51 |
print(decoded_cont)
|
52 |
```
|
|
|
42 |
num_beams=8,
|
43 |
no_repeat_ngram_size=2,
|
44 |
early_stopping=True,
|
45 |
+
pad_token_id= tokenizer.pad_token_id,
|
46 |
eos_token_id = tokenizer.eos_token_id)
|
47 |
|
48 |
# Decode Poet
|
49 |
+
decoded_cont = tokenizer.decode(out[0], skip_special_tokens=True)
|
50 |
|
51 |
print(decoded_cont)
|
52 |
```
|