Update app.py

app.py

@@ -5,18 +5,29 @@ from transformers import GPT2LMHeadModel, GPT2Tokenizer
 model_name = "gpt2"
 tokenizer = GPT2Tokenizer.from_pretrained(model_name)
 model = GPT2LMHeadModel.from_pretrained(model_name)
+def make_prompt(example_title, example_content, new_title):
+    prompt = f"""
+    Title: The Benefits of Daily Meditation
 
+    Blog Content:
+    Meditation is a practice where an individual uses a technique – such as mindfulness, 
+or focusing the mind on a particular object, thought, or activity – to train attention and awareness, 
+and achieve a mentally clear and emotionally calm and stable state. Daily meditation can bring numerous 
+benefits such as reducing stress, improving concentration, and promoting a healthy lifestyle.
+
+    Title: {new_title}
+
+    Blog Content:
+    """
+    return prompt
 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.")
 
-blog_content = generate_blog(title)
-st.subheader("Generated Blog")
-st.write(blog_content)
+if st.button("Generate Blog"):
+    if title:
+        title=make_prompt(title)
+        blog_content = generate_blog(title)
+        st.subheader("Generated Blog")
+        st.write(blog_content)