Update app.py

app.py

@@ -9,8 +9,8 @@ import tempfile
 # —––––––– Requirements —–––––––
 # streamlit>=1.20
 # pillow>=9.0
-# transformers>=4.30
 # torch>=2.0.0
+# transformers>=4.30
 # sentencepiece>=0.1.97
 # gTTS>=2.3.1
 
@@ -48,9 +48,8 @@ captioner, storyteller = load_pipelines()
 # —––––––– Main App —–––––––
 uploaded = st.file_uploader("Upload an image:", type=["jpg", "png", "jpeg"])
 if uploaded:
-    # Display uploaded image
+    # Load and display the image
     img = Image.open(uploaded).convert("RGB")
-    img.thumbnail((2048, 2048), Image.ANTIALIAS)
     st.image(img, use_container_width=True)
 
     # Generate caption
@@ -62,7 +61,7 @@ if uploaded:
         st.stop()
     st.success(f"**Caption:** {caption}")
 
-    # Generate story
+    # Build prompt and generate story
     prompt = (
         f"Image description: {caption}\n"
         "Write a coherent, 50-100 word children’s story that flows naturally."
@@ -79,6 +78,7 @@ if uploaded:
     if len(words) > 100:
         story = " ".join(words[:100]) + ("" if story.endswith('.') else ".")
 
+    # Display story
     st.subheader("📚 Your Magical Story")
     st.write(story)
 
@@ -95,3 +95,4 @@ if uploaded:
 # Footer
 st.markdown("---\n*Made with ❤️ by your friendly story wizard*")
 
+