Update app.py

app.py

@@ -23,14 +23,18 @@ def imgquery(payload):
     response = requests.post(API_URL_IMG, headers=headers, json=payload)
     return response.content
 
+from IPython.display import Audio
+
 def generate_audio(prompt):
     audio_bytes = query({"inputs": prompt})
-    
-    with open("output_audio.mp3", "wb") as f:
+
+    with open("output_audio.wav", "wb") as f:
         f.write(audio_bytes)
-        
+    
     display(Audio(data=audio_bytes, autoplay=False))
-    return audio_bytes
+    
+    return "output_audio.wav"
+
 
 def generate_image(prompt):
     image_bytes = imgquery({
@@ -57,12 +61,12 @@ if generate_btn:
     with audio1:
         st.image(cover1)
         st.header(title1)
-        st.audio(audio1)
+        st.audio(audio1, format="audio/wav")
     with audio2:
         st.image(cover2)
         st.header(title2)
-        st.audio(audio2)
+        st.audio(audio2, format="audio/wav")
     with audio3:
         st.image(cover3)
         st.header(title3)
-        st.audio(audio3)
+        st.audio(audio3, format="audio/wav")