Update app.py

app.py

@@ -11,18 +11,18 @@ def img2text(url):
 
 # text2story
 def text2story(text):
-    pipe = pipeline("text-generation", model="openai-community/gpt2")
+    pipe = pipeline("text-generation", model="pranavpsv/gpt2-story-gen")
     story_text = pipe(text, max_length=100, num_return_sequences=1)[0]['generated_text']
     return story_text
 
 # text2audio
 def text2audio(story_text):
-    pipe = pipeline("text-to-speech", model="facebook/fastspeech2-en-ljspeech")
+    pipe = pipeline("text-to-audio", model="Matthijs/mms-tts-nld")
     audio_data = pipe(story_text)
     return audio_data
 
 # main part
-def main():
+# def main():
     st.set_page_config(page_title="Your Image to Audio Story",
                    page_icon="🦜")
     st.header("Turn Your Image to Audio Story")
@@ -58,5 +58,5 @@ def main():
                      sample_rate = audio_data['sampling_rate'])
 
 # Run the main function
-if __name__ == "__main__":
-    main()
+# if __name__ == "__main__":
+#     main()