Update app.py

app.py

@@ -9,37 +9,36 @@ from gtts import gTTS
 import tempfile
 
 # —––––––– Page Config —–––––––
-st.set_page_config(page_title="Magic Story Generator (CPU)", layout="centered")
-st.title("📖✨ Turn Images into Children's Stories (CPU)")
+st.set_page_config(page_title="Magic Story Generator", layout="centered")
+st.title("📖✨ Turn Images into Children's Stories")
 
 # —––––––– Clients (cached) —–––––––
 @st.cache_resource(show_spinner=False)
 def load_clients():
     hf_token = st.secrets["HF_TOKEN"]
 
-    # Authenticate once so pipelines use your token automatically
+    # Authenticate for both HF Hub and transformers
     os.environ["HUGGINGFACEHUB_API_TOKEN"] = hf_token
     login(hf_token)
 
-    # Pin cache locally to avoid re-downloads
+    # Pin transformers cache locally via env var
     cache_dir = "./hf_cache"
     os.makedirs(cache_dir, exist_ok=True)
     os.environ["TRANSFORMERS_CACHE"] = cache_dir
 
-    # 1) BLIP-based image captioning client
+    # 1) BLIP image-captioning client
     caption_client = InferenceApi(
         repo_id="Salesforce/blip-image-captioning-base",
         token=hf_token
     )
 
-    # 2) Text-generation pipeline forced onto CPU
+    # 2) Text-generation pipeline on CPU (no cache_dir arg here!)
     t0 = time.time()
     story_generator = pipeline(
         task="text-generation",
         model="deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B",
         tokenizer="deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B",
-        device=-1,             # CPU only
-        cache_dir=cache_dir
+        device=-1  # force CPU
     )
     st.text(f"✅ Story model loaded in {time.time() - t0:.1f}s (cached thereafter)")
 
@@ -65,15 +64,15 @@ def generate_caption(img: Image.Image) -> str:
 def generate_story(caption: str) -> str:
     prompt = f"""
 You are a creative children’s-story author.
-Below is the description of an image:
+Below is an image description:
 “{caption}”
 
-Write a coherent, 50 to 100-word story that:
-1. Introduces the main character from the image.
+Write a coherent 50–100 word story that:
+1. Introduces the main character.
 2. Shows a simple problem or discovery.
-3. Resolves it in a happy ending.
+3. Has a happy resolution.
 4. Uses clear language for ages 3–8.
-5. Keeps each sentence under 20 words.
+5. Keeps sentences under 20 words.
 Story:
 """
     t0 = time.time()
@@ -86,14 +85,13 @@ Story:
         no_repeat_ngram_size=3,
         do_sample=True
     )
-    gen_time = time.time() - t0
-    st.text(f"⏱ Generated in {gen_time:.1f}s on CPU")
+    st.text(f"⏱ Generated in {time.time() - t0:.1f}s on CPU")
 
     text = outputs[0]["generated_text"].strip()
-    # Remove the echoed prompt portion
+    # strip the prompt echo
     if text.startswith(prompt):
         text = text[len(prompt):].strip()
-    # Enforce max 100 words
+    # enforce ≤100 words
     words = text.split()
     if len(words) > 100:
         text = " ".join(words[:100])
@@ -103,7 +101,7 @@ Story:
 
 
 # —––––––– Main App Flow —–––––––
-uploaded = st.file_uploader("Upload an image:", type=["jpg", "png", "jpeg"])
+uploaded = st.file_uploader("Upload an image:", type=["jpg","png","jpeg"])
 if uploaded:
     img = Image.open(uploaded).convert("RGB")
     if max(img.size) > 2048:
@@ -134,3 +132,4 @@ if uploaded:
 
 # Footer
 st.markdown("---\n*Made with ❤️ by your friendly story wizard*")
+