Update app.py

app.py

@@ -57,19 +57,17 @@ def generate_story(caption, model, tokenizer):
         model: GPT-2 model.
         tokenizer: GPT-2 tokenizer.
     Returns:
-        story: Generated story as a string.
+        story: Generated story as a string (100 words or less).
     """
     # Add a child-friendly story prompt
     story_prompt = (
         f"Once upon a time, in a magical land, {caption}. "
-        "This is the story of what happened next: "
-        "There was a little hero who loved adventures. "
-        "One day, something amazing happened! "
+        "One sunny day, something magical happened! "
     )
     input_ids = tokenizer.encode(story_prompt, return_tensors="pt")
     output = model.generate(
         input_ids,
-        max_length=100,  
+        max_length=100,  # Limit story length to 100 words
         num_return_sequences=1,
         no_repeat_ngram_size=2,
         temperature=0.9,  # Add randomness for creativity