Update app.py

app.py

@@ -76,8 +76,6 @@ def get_story(caption):
     """
     Generates a humorous and engaging children's story based on the caption.
     Uses a prompt to instruct the model and limits token generation to 80 tokens.
-    
-    If no text is generated, a fallback story is returned.
     """
     prompt = (
         f"Write a funny, warm, and imaginative children's story for ages 3-10, 50-100 words, "
@@ -91,12 +89,16 @@ def get_story(caption):
         top_p=0.9,
         return_full_text=False
     )
-    # Log the raw result for debugging (viewable in server logs)
+    
+    # Log the raw result for debugging (this is viewable in the server logs)
     print("Story generation raw result:", result)
     
     raw_story = result[0].get("generated_text", "").strip()
-    if not raw_story:
-        raw_story = "Once upon a time, the park was filled with laughter as children played happily under the bright sun."
+    
+    # Remove the prompt from the output if it is included.
+    if raw_story.startswith(prompt):
+        raw_story = raw_story[len(prompt):].strip()
+    
     words = raw_story.split()
     story = " ".join(words[:100])
     return story