Update app.py

app.py

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