| from transformers import pipeline | |
| text_generator = pipeline("text-generation", model="gpt2") | |
| def generate_ad(platform, audience, product, tone): | |
| prompt = f"Write a {tone} ad for {product} targeting {audience} on {platform}." | |
| result = text_generator(prompt, max_length=80, num_return_sequences=1) | |
| return result[0]['generated_text'] |