Update app.py

app.py

@@ -8,7 +8,8 @@ tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=True)
 
 def generate_blogpost(topic):
     inputs = tokenizer.encode(topic, return_tensors='pt')
-    outputs = model.generate(inputs, max_length=500, num_return_sequences=1)
+    attention_mask = tokenizer.encode_plus(topic, return_tensors='pt')['attention_mask']
+    outputs = model.generate(inputs, attention_mask=attention_mask, max_length=500, num_return_sequences=1, pad_token_id=tokenizer.eos_token_id)
     text = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return text
 
@@ -17,4 +18,4 @@ st.title('Blog Post Generator')
 topic = st.text_input('Enter a topic:')
 if topic:
     blogpost = generate_blogpost(topic)
-    st.write(blogpost)
+    st.write(blogpost)