Update app.py

app.py

@@ -49,8 +49,7 @@ def text2story(caption, _models):
     # Updated prompt includes explicit instructions to incorporate details from the caption
     prompt = (
         "<|system|>\n"
-        "You are a creative story generating assistant that always incorporates key details "
-        "from the provided image caption into your story.\n"
+        "You are a creative story generating assistant that always incorporates key details from the provided image caption into your story.\n"
         "<|user|>\n"
         f"Please generate a short story within 100 words that clearly utilizes the following image caption and expands upon it thoroughly:\n\"{caption}\"\n"
         "<|assistant|>"
@@ -59,7 +58,7 @@ def text2story(caption, _models):
     # Generate story with parameters tuned for creativity and coherence
     response = story_generator(
         prompt,
-        max_new_tokens=120,  # Enough tokens for a complete story
+        max_new_tokens=100,  # Enough tokens for a complete story
         do_sample=True,
         temperature=0.7,     # Balanced creativity
         top_p=0.9,           # Focus on more likely tokens for coherence