Update app.py

app.py

@@ -42,13 +42,14 @@ def text2Speech(story) :
 
     def query(payload):
     	response = requests.post(API_URL, headers=headers, json=payload)
+        with open('audio.flac','wb') as file:
+        	file.write(response.content)
     	return response.json()
 	
     output = query({
 	"inputs": story,
     })
-    with open('audio.flac','wb') as file:
-    	file.write(output)
+
 
 def main() :
     st.header("Turn img into Audio Story")
@@ -64,10 +65,9 @@ def main() :
         with st.expander("scenario"):
             st.write(scenario)
         story = generate_story(scenario)
-        text2Speech(story)
         with st.expander("story"):
             st.write(story)
-
+        text2Speech(story)
         st.audio("audio.flac")