Create generator.py

generator.py

@@ -0,0 +1,8 @@
+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']