skytnt
/

gpt2-japanese-lyric-small

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

skytnt commited on May 12, 2022

Commit

62fed27

•

1 Parent(s): bd8bd0b

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -29,7 +29,7 @@ model = GPT2LMHeadModel.from_pretrained("skytnt/gpt2-japanese-lyric-small")
 def gen_lyric(prompt_text: str):
-    prompt_text = prompt_text.replace("\n", "\\n")
     prompt_tokens = tokenizer.tokenize(prompt_text)
     prompt_token_ids = tokenizer.convert_tokens_to_ids(prompt_tokens)
     prompt_tensor = torch.LongTensor(prompt_token_ids)

 def gen_lyric(prompt_text: str):
+    prompt_text = prompt_text.replace("\n", "\\n ")
     prompt_tokens = tokenizer.tokenize(prompt_text)
     prompt_token_ids = tokenizer.convert_tokens_to_ids(prompt_tokens)
     prompt_tensor = torch.LongTensor(prompt_token_ids)