Update app.py

app.py

@@ -9,6 +9,7 @@ import spaces
 import tempfile
 import os
 import numpy as np
+import io
 
 # Check for CUDA availability and set device
 try:
@@ -48,16 +49,17 @@ def encode(audio_file_path):
         signal = jnp.expand_dims(signal, axis=(0, 1))  # Add batch and channel dimensions
 
         # Set chunk duration based on available GPU memory (adjust as needed)
-        win_duration = 1.0
+        win_duration = 5.0
 
         # Compress using chunking
         dac_file = model.compress(compress_chunk, signal, sample_rate, win_duration=win_duration)
 
-        # Save the compressed DAC file to a file in the current directory
-        output_path = "compressed_audio.dac"
-        dac_file.save(output_path)
+        # Save the compressed DAC file to a BytesIO object
+        buffer = io.BytesIO()
+        dac_file.save(buffer)
+        buffer.seek(0)
         
-        return output_path
+        return (buffer, "compressed_audio.dac")
 
     except Exception as e:
         gr.Warning(f"An error occurred during encoding: {e}")
@@ -103,4 +105,4 @@ with gr.Blocks() as demo:
 
         decode_button.click(decode, inputs=compressed_input, outputs=decoded_output)
 
-demo.queue().launch()
+demo.queue().launch()