You can deploy this model and start earning money today!

huggingartists/
machine-gun-kelly

Views

No views yet

🤖 HuggingArtists Model 🤖

Machine Gun Kelly

@machine-gun-kelly

I was made with huggingartists.

Create your own bot based on your favorite artist with the demo!

How does it work?

To understand how the model was developed, check the W&B report.

Training data

The model was trained on lyrics from Machine Gun Kelly.

Dataset is available here. And can be used with:

python

from datasets import load_dataset

dataset = load_dataset("huggingartists/machine-gun-kelly")

Explore the data, which is tracked with W&B artifacts at every step of the pipeline.

Training procedure

The model is based on a pre-trained GPT-2 which is fine-tuned on Machine Gun Kelly's lyrics.

Hyperparameters and metrics are recorded in the W&B training run for full transparency and reproducibility.

At the end of training, the final model is logged and versioned.

How to use

You can use this model directly with a pipeline for text generation:

python

from transformers import pipeline
generator = pipeline('text-generation',
                     model='huggingartists/machine-gun-kelly')
generator("I am", num_return_sequences=5)

Or with Transformers library:

python

from transformers import AutoTokenizer, AutoModelWithLMHead
  
tokenizer = AutoTokenizer.from_pretrained("huggingartists/machine-gun-kelly")

model = AutoModelWithLMHead.from_pretrained("huggingartists/machine-gun-kelly")

Limitations and bias

The model suffers from the same limitations and bias as GPT-2.

In addition, the data present in the user's tweets further affects the text generated by the model.

About

Built by Aleksey Korshuk

For more details, visit the project repository.