Update app.py

app.py

@@ -45,11 +45,14 @@ tokenizer = AutoTokenizer.from_pretrained("gpt2-large")
 model = AutoModelForCausalLM.from_pretrained("gpt2-large")
 
 def generate_blog(title):
+
+    prompt = f"write a blog about {title}"
     # Encode the input text
-    inputs = tokenizer.encode(title, return_tensors='pt')
+    inputs = tokenizer.encode(prompt, return_tensors='pt')
     
     # Generate the output
-    outputs = model.generate(inputs, max_length=500, num_return_sequences=1, pad_token_id=tokenizer.eos_token_id)
+    # outputs = model.generate(inputs, max_length=500, num_return_sequences=1, pad_token_id=tokenizer.eos_token_id)
+    outputs = model.generate(inputs, max_length=500, num_return_sequences=1, do_sample=True, top_p=0.95, top_k=60)
 
     # Decode the output text
     blog_post = tokenizer.decode(outputs[0], skip_special_tokens=True)