Update app.py

app.py

@@ -23,18 +23,16 @@ 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.wav", "wb") as f:
-        f.write(audio_bytes)
     
-    display(Audio(data=audio_bytes, autoplay=False))
+    bytes_io = io.BytesIO()
+    bytes_io.write(audio_bytes)
+    bytes_io.seek(0)
     
-    return "output_audio.wav"
-
+    display(Audio(data=bytes_io.read(), autoplay=False))
+    
+    return bytes_io
 
 def generate_image(prompt):
     image_bytes = imgquery({