Update app.py

app.py

@@ -18,8 +18,12 @@ def text2story(text):
 
 # text2audio
 def text2audio(story_text):
-    audio_data = pipeline("text-to-speech", model="suno/bark-small")
-    return audio_data
+    text_to_speech_model = pipeline("text-to-speech", model="suno/bark-small")
+    audio_data = text_to_speech_model(story_text)
+    return {
+        'audio': audio_data[0]['audio'],
+        'sampling_rate': audio_data[0]['sampling_rate']
+    }
 
 #main part