File size: 203 Bytes
812dd54
70b1905
812dd54
70b1905
812dd54
 
1
2
3
4
5
6
7
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"]