Update app.py

app.py

@@ -49,7 +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 only gives me a happy story based on image caption given, no reply and quotation needed.\n"
+        "You are a creative story generating assistant that only gives me a happy story based on image caption given, no reply or quotation are needed which means that you don't need to answer but just give me the story with no quotation.\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|>"
@@ -58,7 +58,7 @@ def text2story(caption, _models):
     # Generate story with parameters tuned for creativity and coherence
     response = story_generator(
         prompt,
-        max_new_tokens=250,  # Enough tokens for a complete story
+        max_new_tokens=120,  # 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