Edit model card
🤖 AI CYBORG 🤖
NSW RFS & NSW SES & Fire and Rescue NSW
@frnsw-nswrfs-nswses

I was made with huggingtweets.

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

How does it work?

The model uses the following pipeline.

pipeline

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

Training data

The model was trained on tweets from NSW RFS & NSW SES & Fire and Rescue NSW.

Data NSW RFS NSW SES Fire and Rescue NSW
Tweets downloaded 3250 3248 3249
Retweets 275 2093 875
Short tweets 12 12 48
Tweets kept 2963 1143 2326

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 @frnsw-nswrfs-nswses's tweets.

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:

from transformers import pipeline
generator = pipeline('text-generation',
                     model='huggingtweets/frnsw-nswrfs-nswses')
generator("My dream is", num_return_sequences=5)

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 Boris Dayma

Follow

For more details, visit the project repository.

GitHub stars

Downloads last month
12