Update app.py

app.py

@@ -12,7 +12,7 @@ st.title("GPT-2 Blog Post Generator")
 # Text input for the user
 text = st.text_area("Enter your Topic: ")
 
-if text:
+def generate_text(text):
     try:
         # Encode input text
         encoded_input = tokenizer(text, return_tensors='pt')
@@ -33,10 +33,13 @@ if text:
 
         # Decode generated text
         generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
+        return generate_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}")
-        
+if st.button("Generate"):
+    generated_text = generate_text(text)
+    st.write(generated_text)