Update app.py

app.py

@@ -7,7 +7,14 @@ from PIL import Image
 # Load models
 def load_models():
     image_to_text = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
-    storyteller = pipeline("text2text-generation", model="google/flan-t5-small", max_length=100)  # Max tokens set to 100
+    storyteller = pipeline(
+        "text2text-generation",
+        model="google/flan-t5-xl",
+        max_length=100,
+        temperature=1,
+        no_repeat_ngram_size=2,   # Prevents repeating 2-grams
+        penalty_alpha=0.5         # Penalty for repeated n-grams
+    )
     return image_to_text, storyteller
 
 # Process image to text
@@ -17,7 +24,7 @@ def generate_caption(image, image_to_text):
 
 # Generate a narrative story using an optimized Flan-T5 prompt
 def generate_story(text, storyteller):
-    prompt = f"Write a 75-word creative and engaging short story based on the following description: {text}."
+    prompt = f'''Write a 50-word story about "{text}".'''
     story = storyteller(prompt)
     return story[0]["generated_text"] if story else "No story generated."