Update app.py

app.py

@@ -39,10 +39,18 @@ def Image_Caption(image):
 
 # Function to generate a story
 def Generate_story(textAbout):
-    storyAbout = {"role": "user", "content": f'write a long story about {textAbout} that takes 3 min to read'}
-    story = generator(storyAbout)
+    # Define the prompt as a string
+    prompt = f'write a long story about {textAbout} that takes 3 min to read'
+    
+    # Generate the story using the generator pipeline
+    story = generator(prompt)
+    
+    # Extract the generated text
     story = story[0]['generated_text']
+    
+    # Clean up the story
     story = story.replace('\n', ' ').replace('arafed', ' ')
+    
     return story
 
 # Function to generate audio