blog-writer-ai / app.py
Pooja P
used transformer instead of openai
812dd54
raw
history blame
203 Bytes
from transformers import pipeline
generator = pipeline("text-generation", model="gpt2")
def generate_blog(topic):
return generator(f"Write a blog on: {topic}", max_length=200)[0]["generated_text"]