gpt2-medium model fine-tuned to generate roasts.1from transformers import pipeline
2
3generator = pipeline('text-generation', model='3naruto3/roast-bot')
4prompt = "Roast: My terrible code"
5result = generator(prompt, max_length=50)
6print(result[0]['generated_text'])