Update app.py

app.py

@@ -4,23 +4,26 @@ from PIL import Image
 
 # Creates a brief description of a picture using a smart model
 def generate_caption(image):
-    # Loads the BLIP model to examine and describe the picture
-    image_to_text = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
-    caption = image_to_text(image)[0]["generated_text"]
+    with st.spinner("🔍 Looking at your picture..."):
+        # Loads the BLIP model to examine and describe the picture
+        image_to_text = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
+        caption = image_to_text(image)[0]["generated_text"]
     return caption
 
 # Builds a story from the picture’s description
 def generate_story(caption):
-    # Uses a story model to craft a tale from the description
-    pipe = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2")
-    story = pipe(caption)[0]['generated_text']
+    with st.spinner("✍️ Writing a fun story..."):
+        # Uses a story model to craft a tale from the description
+        pipe = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2")
+        story = pipe(caption)[0]['generated_text']
     return story
 
 # Turns the story into spoken audio
 def generate_audio(story):
-    # Uses a speech model to read the story aloud
-    pipe = pipeline("text-to-speech", model="facebook/mms-tts-eng")
-    audio = pipe(story)
+    with st.spinner("🎙️ Turning story into audio..."):
+        # Uses a speech model to read the story aloud
+        pipe = pipeline("text-to-speech", model="facebook/mms-tts-eng")
+        audio = pipe(story)
     return audio
 
 # Streamlit UI: Makes a simple interface for kids to enjoy stories