llmahmad commited on
Commit
db7ea0e
·
verified ·
1 Parent(s): a6d02b0

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +26 -0
app.py ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Load the GPT-2 model and tokenizer
2
+ model_name = 'gpt2-large'
3
+ tokenizer = GPT2Tokenizer.from_pretrained(model_name)
4
+ model = GPT2LMHeadModel.from_pretrained(model_name)
5
+
6
+ def generate_blog_post(topic):
7
+ # Encode the input topic
8
+ inputs = tokenizer.encode(topic, return_tensors='pt')
9
+
10
+ # Generate the blog post
11
+ outputs = model.generate(inputs, max_length=500, num_return_sequences=1, no_repeat_ngram_size=2,
12
+ do_sample=True, top_k=50, top_p=0.95, temperature=0.9)
13
+
14
+ # Decode the generated text
15
+ blog_post = tokenizer.decode(outputs[0], skip_special_tokens=True)
16
+ return blog_post
17
+
18
+ # Streamlit app
19
+ st.title("Blog Post Generator")
20
+ st.write("Enter a topic to generate a blog post.")
21
+
22
+ topic = st.text_input("Topic:")
23
+
24
+ if st.button("Generate"):
25
+ blog_post = generate_blog_post(topic)
26
+ st.write(blog_post)