Update app.py

app.py

@@ -22,7 +22,7 @@ _image_caption_pipeline = pipeline(
 )
 
 # Global model configuration constants
-_text_generation_pipeline = pipeline("text-generation", model="Qwen/Qwen1.5-0.5B",max_new_tokens=100)
+_text_generation_pipeline = pipeline("text-generation", model="Qwen/Qwen3-1.7B")
 
 # Initialize TTS components once to avoid reloading
 _SPEECH_PIPELINE = pipeline("text-to-speech", model="microsoft/speecht5_tts")
@@ -82,15 +82,14 @@ def generate_story_content(system_prompt: str, user_prompt: str) -> str:
     try:
         # Prepare chat message structure
         conversation_history = [
-            {"role": "system", "content": system_prompt},
-            {"role": "user", "content": user_prompt},
+            {"role": "user", "content": system_prompt+user_prompt+"/no_think"},
         ]
 
         # Generate the story
         story=_text_generation_pipeline(conversation_history)
 
         # Extract the stroy result
-        stroy_result=story[0]["generated_text"][2]["content"]
+        stroy_result=story[0]["generated_text"][1]["content"][19:]
         
         # Process and clean output
         return stroy_result
@@ -259,15 +258,15 @@ if uploaded_image is not None:
         # Define prompt templates
         PROMPT_TEMPLATES = {
             "educational": {
-                "system": "You are a children's educator. Create a simple 100-word story that teaches basic life skills or moral lessons.",
+                "system": "You are a children's educator writer. Create an educational story about ",
                 "icon": "📚"
             },
             "adventure": {
-                "system": "You are a fantasy writer. Create a 100-word magical adventure story suitable for children.",
+                "system": "You are a fantasy writer. Create a adventure story about ",
                 "icon": "🌠"
             },
             "animal": {
-                "system": "You are an animal expert. Create a 100-word story about friendly animals learning together.",
+                "system": "You are an animal themes writer. Create a animal fairy tales about ",
                 "icon": "🐻"
             }
         }