Adding Prompt

app.py

@@ -11,8 +11,10 @@ def generate_blog(title, model_name='gpt2', max_length=500):
     tokenizer = GPT2Tokenizer.from_pretrained(model_name)
     model = GPT2LMHeadModel.from_pretrained(model_name).to(device)
 
+    prompt = f"Write a blog post based on this Title: {title}"
+
     # Prepare the input
-    input_ids = tokenizer.encode(title, return_tensors='pt').to(device)
+    input_ids = tokenizer.encode(prompt, return_tensors='pt').to(device)
 
     # Generate text
     output = model.generate(input_ids, max_length=max_length, num_return_sequences=1, no_repeat_ngram_size=2, early_stopping=True)