Update app.py

app.py

@@ -9,6 +9,13 @@ import torch
 from gtts import gTTS
 import tempfile
 
+# —––––––– Requirements —–––––––
+# This app uses a Hugging Face Transformers version that supports
+# the Qwen2.5-Omni architecture via `trust_remote_code`.
+# Install using:
+# pip install git+https://github.com/huggingface/transformers.git
+# and the rest of the requirements listed at the end.
+
 # —––––––– Page Config —–––––––
 st.set_page_config(page_title="Magic Story Generator (Qwen2.5)", layout="centered")
 st.title("📖✨ Turn Images into Children's Stories (Qwen2.5-Omni-7B)")
@@ -27,7 +34,7 @@ def load_clients():
         token=hf_token
     )
 
-    # 2) Load Qwen2.5-Omni model & tokenizer
+    # 2) Load Qwen2.5-Omni model & tokenizer via remote code
     t0 = time.time()
     tokenizer = AutoTokenizer.from_pretrained(
         "Qwen/Qwen2.5-Omni-7B",
@@ -40,7 +47,6 @@ def load_clients():
         torch_dtype=torch.bfloat16,
         attn_implementation="flash_attention_2"
     )
-    # 3) Build text2text pipeline
     storyteller = pipeline(
         task="text2text-generation",
         model=model,
@@ -85,12 +91,9 @@ def generate_story(caption: str) -> str:
     st.text(f"⏱ Generated in {gen_time:.1f}s on GPU/CPU")
 
     story = result[0]["generated_text"].strip()
-    # Enforce ≤100 words
     words = story.split()
     if len(words) > 100:
-        story = " ".join(words[:100])
-        if not story.endswith('.'):
-            story += '.'
+        story = " ".join(words[:100]) + ('.' if not story.endswith('.') else '')
     return story
 
 # —––––––– Main App —–––––––