Views
No views yet
1from transformers import GPT2LMHeadModel, GPT2Tokenizer
2
3tokenizer = GPT2Tokenizer.from_pretrained("SpartanCinder/GPT2-pretrained-lyric-generation")
4model = GPT2LMHeadModel.from_pretrained("SpartanCinder/GPT2-pretrained-lyric-generation")
5
6input_ids = tokenizer.encode("Once upon a time", return_tensors='pt')
7output = model.generate(input_ids, max_length=100, num_return_sequences=5, do_sample=True, top_p=0.9)
8print(tokenizer.decode(output[0], skip_special_tokens=True))