Update app.py

app.py

@@ -21,17 +21,17 @@ def decompress_chunk(c):
     return model.apply(variables, c, method='decompress_chunk')
 
 @spaces.GPU
-def encode(audio_file_path):
+def encode(audio_file):
     try:
-        # Load a mono audio file
-        signal, sample_rate = librosa.load(audio_file_path, sr=44100, mono=True) 
+        # Load a mono audio file directly from BytesIO
+        signal, sample_rate = librosa.load(audio_file, sr=44100, mono=True) 
 
         signal = jnp.array(signal, dtype=jnp.float32)
         while signal.ndim < 3:
             signal = jnp.expand_dims(signal, axis=0)
 
         # Set chunk duration based on available GPU memory (adjust as needed)
-        win_duration = 0.5  # You might need to experiment with this value
+        win_duration = 5.0  # Start with 5 seconds
 
         # Compress using chunking
         dac_file = model.compress(compress_chunk, signal, sample_rate, win_duration=win_duration)