Multiple file formats

app.py

@@ -20,10 +20,10 @@ class DACApi:
 
     @spaces.GPU
     def encode_audio(self, input_file):
-        # Convert MP3 to WAV if necessary
-        if input_file.name.lower().endswith('.mp3'):
-            print("Converting MP3 to WAV...")
-            audio = AudioSegment.from_mp3(input_file.name)
+        # Convert various audio formats to WAV if necessary
+        if not input_file.name.lower().endswith('.wav'):
+            print(f"Converting {input_file.name} to WAV...")
+            audio = AudioSegment.from_file(input_file.name)
             input_wav = io.BytesIO()
             audio.export(input_wav, format="wav")
             input_wav.seek(0)
@@ -67,7 +67,7 @@ class DACApi:
         print("Decompressing audio...")
         decompressed = self.model.decompress(compressed)
 
-        audio_data = decompressed.audio_data.cpu().numpy().squeeze().T
+        audio_data = decompressed.audio_data.cpu().detach().numpy().squeeze().T
         sample_rate = decompressed.sample_rate
 
         return (sample_rate, audio_data)