Update app.py

app.py

@@ -1,5 +1,7 @@
 import streamlit as st
 from transformers import pipeline
+import soundfile as sf
+import io
 
 # --- Load models once at the beginning ---
 image_to_text_model = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
@@ -25,6 +27,13 @@ def text2audio(story_text):
     speech_data = story_to_audio_model(story_text)
     return speech_data
 
+# play_audio
+def play_audio(audio_data):
+    audio_buffer = io.BytesIO()
+    sf.write(audio_buffer, audio_data['audio'], audio_data['sampling_rate'], format='WAV')
+    audio_buffer.seek(0)
+    st.audio(audio_buffer, format='audio/wav')
+
 # --- Streamlit App ---
 st.set_page_config(page_title="Your Image to Audio Story", page_icon="🦜")
 st.header("Turn Your Image to Audio Story")