Update app.py

app.py

@@ -22,10 +22,10 @@ if st.button("Generate Blog Post"):
         inputs_encoded = tokenizer.encode(prompt, return_tensors="pt")
 
         # Generate text
-        model_output = model.generate(inputs_encoded, max_new_tokens=500, do_sample=True, temperature=0.7)
+        model_output = model.generate(inputs_encoded, max_new_tokens=200, do_sample=True, temperature=0.7)
         
         # Decode the output
-        output = tokenizer.decode(model_output[0], skip_special_tokens=True)
+        output = tokenizer.decode(model_output, skip_special_tokens=True)
         
         # Display the generated blog post
         st.subheader("Generated Blog Post:")