Update app.py

app.py

@@ -19,27 +19,29 @@ def generate_text(text):
 
         # Generate text
         output = model.generate(
-	        input_ids=encoded_input,
-	        max_length=100,  # Specify the max length for the generated text
-	        num_return_sequences=1,  # Number of sequences to generate
-	        no_repeat_ngram_size=2,  # Avoid repeating n-grams of length 2
-	        top_k=50,  # Limits the sampling pool to top_k tokens
-	        top_p=0.95,  # Cumulative probability threshold for nucleus sampling
-	        temperature=0.7,  # Controls the randomness of predictions
-	        do_sample=True,  # Enable sampling
-	        attention_mask=encoded_input.new_ones(encoded_input.shape),
-	        pad_token_id=tokenizer.eos_token_id  # Use the end-of-sequence token as padding
-	    )
+            input_ids=encoded_input['input_ids'],
+            max_length=100,  # Specify the max length for the generated text
+            num_return_sequences=1,  # Number of sequences to generate
+            no_repeat_ngram_size=2,  # Avoid repeating n-grams of length 2
+            top_k=50,  # Limits the sampling pool to top_k tokens
+            top_p=0.95,  # Cumulative probability threshold for nucleus sampling
+            temperature=0.7,  # Controls the randomness of predictions
+            do_sample=True,  # Enable sampling
+            attention_mask=encoded_input['attention_mask'],  # Correct attention mask
+            pad_token_id=tokenizer.eos_token_id  # Use the end-of-sequence token as padding
+        )
 
         # Decode generated text
         generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
-        return generate_text
+        return generated_text
 
-        # Display the generated text
-        st.subheader("Generated Blog Post")
-        st.write(generated_text)
     except Exception as e:
         st.error(f"An error occurred: {e}")
+        return None
+
 if st.button("Generate"):
     generated_text = generate_text(text)
-    st.write(generated_text)
+    if generated_text:
+        # Display the generated text
+        st.subheader("Generated Blog Post")
+        st.write(generated_text)