Create your own bot based on your favorite artist with
the demo!
To understand how the model was developed, check the
W&B report.
The model was trained on lyrics from Peter, Paul and Mary.
1from datasets import load_dataset
2
3dataset = load_dataset("huggingartists/peter-paul-and-mary")
Explore the data, which is tracked with
W&B artifacts at every step of the pipeline.
The model is based on a pre-trained
GPT-2 which is fine-tuned on Peter, Paul and Mary's lyrics.
Hyperparameters and metrics are recorded in the
W&B training run for full transparency and reproducibility.
1from transformers import pipeline
2generator = pipeline('text-generation',
3 model='huggingartists/peter-paul-and-mary')
4generator("I am", num_return_sequences=5)
1from transformers import AutoTokenizer, AutoModelWithLMHead
2
3tokenizer = AutoTokenizer.from_pretrained("huggingartists/peter-paul-and-mary")
4
5model = AutoModelWithLMHead.from_pretrained("huggingartists/peter-paul-and-mary")
In addition, the data present in the user's tweets further affects the text generated by the model.
For more details, visit the project repository.