Update app.py

app.py

@@ -11,18 +11,19 @@ def img2text(url):
 # text2story
 def text2story(text):
     #story_text = ""   # to be completed
-    messages = [
-    {"role": "user", "content": "Generate a 100-word story based on the following keywords："+text },
-    ]
-    text_to_story_model= pipeline("text-generation", model="microsoft/Phi-4-mini-instruct", trust_remote_code=True)
+    #messages = [
+    #{"role": "user", "content": "Generate a 100-word story based on the following keywords："+text },
+    #]
+    messages = "Generate a 100-word story based on the following keywords：" + text 
+    text_to_story_model= pipeline("text2text-generation", model="google/flan-t5-small")
     story_text = text_to_story_model(messages)[0]['generated_text']
     return story_text
     
 # text2audio
 def text2audio(story_text):
-    text_to_video_model = pipeline("text-to-speech", model="facebook/mms-tts-eng")
+    text_to_audio_model = pipeline("text-to-speech", model="facebook/mms-tts-eng")
     #audio_data = ""     # to be completed
-    audio_data = text_to_video_model(story_text)
+    audio_data = text_to_audio_model(story_text)
     return audio_data
 
 st.set_page_config(page_title="Your Image to Audio Story",
@@ -49,8 +50,8 @@ if uploaded_file is not None:
     st.write(story)
 
     #Stage 3: Story to Audio data
-    #st.text('Generating audio data...')
-    #audio_data =text2audio(story)
+    st.text('Generating audio data...')
+    audio_data =text2audio(story)
 
     # Play button
     if st.button("Play Audio"):