bump bitrate

app.py

@@ -21,9 +21,9 @@ def numpy_to_mp3(audio_array, sampling_rate):
         channels=1
     )
 
-    # Export the audio segment to MP3 bytes
+    # Export the audio segment to MP3 bytes - use a high bitrate to maximise quality
     mp3_io = io.BytesIO()
-    audio_segment.export(mp3_io, format="mp3")
+    audio_segment.export(mp3_io, format="mp3", bitrate="320k")
 
     # Get the MP3 bytes
     mp3_bytes = mp3_io.getvalue()
@@ -48,7 +48,7 @@ def stream(audio, chunk_length_s):
         if idx == 0:
             first_time = round(time.time() - start_time, 2)
         run_time = round(time.time() - start_time, 2)
-        yield (sampling_rate, chunk), chunk_mp3, first_time, run_time
+        yield chunk_mp3, first_time, run_time
 
 with gr.Blocks() as demo:
     with gr.Row():
@@ -57,11 +57,10 @@ with gr.Blocks() as demo:
             chunk_length = gr.Slider(minimum=2, maximum=10, value=2, step=2, label="Chunk length (s)")
             run_button = gr.Button("Stream audio")
         with gr.Column():
-            audio_out = gr.Audio(streaming=True, autoplay=True, label="wav")
-            audio_out_mp3 = gr.Audio(streaming=True, autoplay=True, format="mp3", label="mp3")
+            audio_out = gr.Audio(streaming=True, autoplay=True, format="mp3", label="mp3")
             first_time = gr.Textbox(label="Time to first chunk (s)")
             run_time = gr.Textbox(label="Time to current chunk (s)")
 
-    run_button.click(fn=stream, inputs=[audio_in, chunk_length], outputs=[audio_out, audio_out_mp3, first_time, run_time])
+    run_button.click(fn=stream, inputs=[audio_in, chunk_length], outputs=[audio_out, first_time, run_time])
 
 demo.launch()