|
|
--- |
|
|
license: mit |
|
|
tags: |
|
|
- text-generation |
|
|
- humor |
|
|
- jokes |
|
|
- distilgpt2 |
|
|
pipeline_tag: text-generation |
|
|
--- |
|
|
|
|
|
# Jokes Model |
|
|
|
|
|
A fine-tuned DistilGPT2 model that generates short, clean, and (sometimes) funny jokes! |
|
|
|
|
|
## Model Details |
|
|
- **Model type:** Causal Language Model (DistilGPT2) |
|
|
- **Fine-tuned on:** 10,000 filtered jokes from [shortjokes.csv](https://www.kaggle.com/datasets/abhinavmoudgil95/short-jokes) |
|
|
- **Training epochs:** 5 |
|
|
- **Max joke length:** 80 tokens |
|
|
|
|
|
## Usage |
|
|
|
|
|
### Direct Inference |
|
|
|
|
|
```python |
|
|
from transformers import pipeline, AutoTokenizer |
|
|
import torch |
|
|
|
|
|
#Please add the BLOCKLIST for clean jokes |
|
|
BLOCKLIST = [ |
|
|
"sex", "naked", "porn", "fuck", "dick", "penis", "ass", |
|
|
"blowjob", "orgasm", "rape", "kill", "die", "shit", |
|
|
"crap", "bastard", "hell", "damn", "bitch", "underage", |
|
|
"pedo", "hit", "shot", "gun", "drug", "drunk", "fag", "cunt" |
|
|
] |
|
|
|
|
|
def is_safe(text): |
|
|
text_lower = text.lower() |
|
|
return not any(bad_word in text_lower for bad_word in BLOCKLIST) |
|
|
|
|
|
def generate_joke(prompt="Tell me a clean joke:"): |
|
|
joke_gen = pipeline( |
|
|
"text-generation", |
|
|
model="FaisalGh/jokes-model", |
|
|
device=0 if torch.cuda.is_available() else -1 |
|
|
) |
|
|
|
|
|
output = joke_gen( |
|
|
prompt, |
|
|
max_length=80, |
|
|
temperature=0.7, |
|
|
top_k=50, |
|
|
top_p=0.9, |
|
|
repetition_penalty=1.5, |
|
|
no_repeat_ngram_size=2, |
|
|
do_sample=True, |
|
|
pad_token_id=50256, |
|
|
eos_token_id=50256 |
|
|
) |
|
|
|
|
|
generated_text = output[0]['generated_text'] |
|
|
first_sentence = generated_text.split(".")[0] + "." |
|
|
|
|
|
return "[Content filtered] Please try again." if not is_safe(first_sentence) else first_sentence.strip() |
|
|
|
|
|
print(generate_joke("Tell me a clean joke:")) |
