Update app.py

app.py

@@ -27,7 +27,7 @@ def text_to_audio(text, tts_model):
 # Function to process the uploaded image and generate a story
 def process_image(image, caption_model):
     # Generate a caption from the uploaded image
-    caption = caption_model(image)[0]['caption']  # Get the caption from the model
+    caption = caption_model(image)[0]['generated_text']  # Get the caption from the model
     # Generate a story from the caption
     story = generate_story(caption)  # Call the story generation function
     return caption, story  # Return both caption and story