Update README.md
Browse files
README.md
CHANGED
@@ -29,7 +29,7 @@ model = GPT2LMHeadModel.from_pretrained("skytnt/gpt2-japanese-lyric-small")
|
|
29 |
|
30 |
|
31 |
def gen_lyric(prompt_text: str):
|
32 |
-
prompt_text = prompt_text.replace("\n", "\\n")
|
33 |
prompt_tokens = tokenizer.tokenize(prompt_text)
|
34 |
prompt_token_ids = tokenizer.convert_tokens_to_ids(prompt_tokens)
|
35 |
prompt_tensor = torch.LongTensor(prompt_token_ids)
|
|
|
29 |
|
30 |
|
31 |
def gen_lyric(prompt_text: str):
|
32 |
+
prompt_text = prompt_text.replace("\n", "\\n ")
|
33 |
prompt_tokens = tokenizer.tokenize(prompt_text)
|
34 |
prompt_token_ids = tokenizer.convert_tokens_to_ids(prompt_tokens)
|
35 |
prompt_tensor = torch.LongTensor(prompt_token_ids)
|