Update app.py

app.py

@@ -30,7 +30,7 @@ def load_clients():
 caption_client, story_client = load_clients()
 
 # —––––––– Main UI
-uploaded = st.file_uploader("Upload an image:", type=["jpg","jpeg","png"])
+uploaded = st.file_uploader("Upload an image:", type=["jpg", "jpeg", "png"])
 if not uploaded:
     st.info("Please upload a JPG/PNG image to begin.")
 else:
@@ -44,7 +44,7 @@ else:
         img.save(buf, format="PNG")
         cap_out = caption_client(data=buf.getvalue())
 
-        # Correctly extract from list/dict
+        # Unwrap list/dict properly
         if isinstance(cap_out, list) and cap_out:
             cap_text = cap_out[0].get("generated_text", "").strip()
         elif isinstance(cap_out, dict):
@@ -53,12 +53,12 @@ else:
             cap_text = str(cap_out).strip()
 
     if not cap_text:
-        st.error("😕 I couldn’t generate a caption. Try uploading a different image.")
+        st.error("😕 Couldn’t generate a caption. Try another image.")
         st.stop()
 
     st.markdown(f"**Caption:** {cap_text}")
 
-    # 3) Build prompt for story
+    # 3) Build prompt
     prompt = (
         f"Here’s an image description: “{cap_text}”.\n\n"
         "Write an 80–100 word playful story for 3–10 year-old children that:\n"
@@ -68,11 +68,11 @@ else:
         "Story:"
     )
 
-    # 4) Generate story
+    # 4) Generate story via HF Inference API (use `params`)
     with st.spinner("✍️ Generating story..."):
         story_out = story_client(
             inputs=prompt,
-            parameters={        # must be `parameters`, not `params`
+            params={             # ← must be `params`, not `parameters`
                 "max_new_tokens": 120,
                 "do_sample": True,
                 "temperature": 0.7,
@@ -90,7 +90,7 @@ else:
             story = str(story_out).strip()
 
     if not story:
-        st.error("😕 I couldn’t generate a story. Please try again!")
+        st.error("😕 Couldn’t generate a story. Please try again!")
         st.stop()
 
     st.markdown("**Story:**")