Update app.py

app.py

@@ -1,7 +1,5 @@
 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")
@@ -29,10 +27,7 @@ def text2audio(story_text):
 
 # 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')
+    st.audio(audio_data['audio'], format='audio/wav', sample_rate=audio_data['sampling_rate'])
 
 # --- Streamlit App ---
 st.set_page_config(page_title="Your Image to Audio Story", page_icon="🦜")