Update app.py

app.py

@@ -39,19 +39,11 @@ def Image_Caption(image):
 
 # Function to generate a story
 def Generate_story(textAbout):
-    # 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
+  storyAbout =  {"role": "user", "content": f'write a long story about {textAbout} that takes 3 min to read'},    
+  story = generator(storyAbout)
+  story = story[0]['generated_text']
+  story = story.replace('\n', ' ').replace('arafed', ' ')
+  return story
 
 # Function to generate audio
 def Generate_audio(text, voice='bm_lewis', speed=1):