Update app.py

app.py

@@ -8,7 +8,7 @@
 
 import streamlit as st
 from transformers import pipeline
-
+from IPython.display import Audio
 
 # function part - FOUR functions
 # img2text()
@@ -30,10 +30,7 @@ def text2audio(story_text):
     audio_data = pipe(story_text)
     return audio_data
 
-# main()
 
-if __name__ == "__main__":
-    main()
 
 def main():
 
@@ -70,3 +67,11 @@ def main():
                         format="audio/wav",
                         start_time=0,
                         sample_rate = audio_data['sampling_rate'])
+
+
+# main()
+
+if __name__ == "__main__":
+    main()
+
+