Spaces:
Sleeping
Sleeping
File size: 811 Bytes
50160f2 4e630e1 50160f2 4e630e1 50160f2 4e630e1 52101ba 4e630e1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
import streamlit as st
from transformers import GPT2LMHeadModel, GPT2Tokenizer
model_name = "gpt2"
tokenizer = GPT2Tokenizer.from_pretrained(model_name)
model = GPT2LMHeadModel.from_pretrained(model_name)
def generate_blog(title):
input_ids = tokenizer.encode(title, return_tensors='pt')
output = model.generate(input_ids, max_length=800, num_return_sequences=1, no_repeat_ngram_size=2)
blog = tokenizer.decode(output[0], skip_special_tokens=True)
return blog
st.title("AI Blog Generator")
st.write("Enter a blog title and the AI will generate the blog content for you.")
# Input blog title
title = st.text_input("Blog Title", "")
if st.button("Generate Blog"):
if title:
blog_content = generate_blog(title)
st.subheader("Generated Blog")
st.write(blog_content)
|