Update app.py

app.py

@@ -11,23 +11,22 @@ def img2text(url):
 
 # text2story
 def text2story(text):
-    pipe = pipeline("text-generation", model="pranavpsv/gpt2-story-gen")
-    story_text = pipe(text, max_length=100, num_return_sequences=1)[0]['generated_text']
+    pipe = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2")
+    story_text = pipe(text)[0]['generated_text']
     return story_text
 
 # text2audio
 def text2audio(story_text):
-    pipe = pipeline("text-to-audio", model="Matthijs/mms-tts-nld")
+    pipe = pipeline("text-to-audio", model="Matthijs/mms-tts-eng")
     audio_data = pipe(story_text)
     return audio_data
 
-# main part
-# def main():
-    st.set_page_config(page_title="Your Image to Audio Story",
-                   page_icon="🦜")
+
+def main():
+    st.set_page_config(page_title="Your Image to Audio Story", page_icon="🦜")
     st.header("Turn Your Image to Audio Story")
     uploaded_file = st.file_uploader("Select an Image...")
-    
+
     if uploaded_file is not None:
         print(uploaded_file)
         bytes_data = uploaded_file.getvalue()
@@ -57,6 +56,6 @@ def text2audio(story_text):
                      start_time=0, 
                      sample_rate = audio_data['sampling_rate'])
 
-# Run the main function
-# if __name__ == "__main__":
-#     main()
+
+if __name__ == "__main__":
+    main()