Update app.py

app.py

@@ -1,25 +1,25 @@
 import streamlit as st
 from transformers import pipeline
 
-# Initialize the Hugging Face pipeline for text generation
+
 generator = pipeline("text-generation", model="gpt2")
 
-# Streamlit app layout
+
 st.title("AI Story Generator")
 
-# Prompt input from the user
+
 prompt = st.text_input("Enter a prompt:")
 
-# Slider to control the maximum length of the generated story
+
 max_length = st.slider("Select the maximum length of the story:", 
                        min_value=50, max_value=500, value=100, step=10)
 
-# Button to trigger the story generation
+
 if st.button("Generate Story"):
     if prompt:
-        # Generate the story based on the prompt and slider value
+        
         result = generator(prompt, max_length=max_length, num_return_sequences=1)
-        # Display the generated story
+        
         st.write(result[0]['generated_text'])
     else:
         st.write("Please enter a prompt to generate a story.")