Update app.py

app.py

@@ -35,7 +35,7 @@ if uploaded_file:
 
     # Step 2: Generate Story (remove prompt from output)
     with st.spinner("Writing a children's story..."):
-        prompt = f"Write a short, imaginative story for children aged 3-10 about this: {caption}"
+        prompt = f"{caption}"
         story_output = gpt2(
             prompt,
             max_length=100,