AhsanShahid commited on
Commit
614359b
1 Parent(s): 386e9b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -3
app.py CHANGED
@@ -20,12 +20,22 @@ topic = st.text_input("Enter the topic for your blog post:")
20
  # Generate blog post button
21
  if st.button("Generate Blog Post"):
22
  if topic:
23
- # Prepare the input for the model
24
- input_text = f"Write a blog post about {topic}."
 
 
25
  inputs = tokenizer.encode(input_text, return_tensors="pt")
26
 
27
  # Generate the blog post using GPT-2 large
28
- outputs = model.generate(inputs, max_length=500, num_return_sequences=1, no_repeat_ngram_size=2, early_stopping=True)
 
 
 
 
 
 
 
 
29
 
30
  # Decode the generated text
31
  blog_post = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
20
  # Generate blog post button
21
  if st.button("Generate Blog Post"):
22
  if topic:
23
+ # Refine the input prompt to guide the model towards generating a blog post
24
+ input_text = f"Write a detailed blog post about {topic}. The post should cover various aspects of the topic and provide valuable information to the readers. Start with an introduction and follow with detailed paragraphs."
25
+
26
+ # Encode the input text
27
  inputs = tokenizer.encode(input_text, return_tensors="pt")
28
 
29
  # Generate the blog post using GPT-2 large
30
+ outputs = model.generate(
31
+ inputs,
32
+ max_length=500,
33
+ num_return_sequences=1,
34
+ no_repeat_ngram_size=2,
35
+ early_stopping=True,
36
+ temperature=0.7,
37
+ top_p=0.9
38
+ )
39
 
40
  # Decode the generated text
41
  blog_post = tokenizer.decode(outputs[0], skip_special_tokens=True)