Update app.py

app.py

@@ -1,4 +1,4 @@
-# Imports - just the essentials
+# Imports
 import streamlit as st
 from transformers import pipeline
 from PIL import Image
@@ -50,13 +50,13 @@ def text2story(caption, _models):
     prompt = f"""<|system|>
 You are a story generating assistant.
 <|user|>
-Please generate a story within 100 words based on: {caption}
+Please generate a story within 100 words based on the image caption: {caption}
 <|assistant|>"""
     
     # Generate story with parameters tuned for brevity and coherence
     response = story_generator(
         prompt,
-        max_new_tokens=100,  # Allow enough tokens for a complete story
+        max_new_tokens=120,  # Allow enough tokens for a complete story
         do_sample=True,
         temperature=0.7,     # Balanced creativity
         top_p=0.9,           # Focus on more likely tokens
@@ -91,7 +91,6 @@ def text2audio(story_text):
 
 # Load models at startup - this happens before the app interface is displayed
 models = load_models()
-st.write("✅ Models loaded and cached!")
 
 # Streamlit app interface
 st.title("Image to Audio Story for Kids")