Update app.py

app.py

@@ -7,7 +7,7 @@ from PIL import Image
 # Load models
 def load_models():
     image_to_text = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
-    storyteller = pipeline("text-generation", model="microsoft/phi-2", max_new_tokens=300)
+    storyteller = pipeline("text2text-generation", model="google/flan-t5-small", max_length=200)
     return image_to_text, storyteller
 
 # Process image to text
@@ -15,10 +15,10 @@ def generate_caption(image, image_to_text):
     result = image_to_text(image)
     return result[0]["generated_text"] if result else "No caption generated."
 
-# Generate a narrative story
+# Generate a narrative story using an optimized Flan-T5 prompt
 def generate_story(text, storyteller):
-    prompt = f"Write a short, engaging story based on this description: {text}"
-    story = storyteller(prompt, do_sample=True, temperature=0.7, max_new_tokens=300)
+    prompt = f"Write a creative and engaging short story based on this description: {text}"
+    story = storyteller(prompt)
     return story[0]["generated_text"] if story else "No story generated."
 
 # Convert text to speech