Modify to output wav at 24k

app.py

@@ -79,7 +79,7 @@ def main(args):
         )
         # Return the audio file path
         if result and len(result) > 0:
-            return result[0]  # Return first audio output
+            return result[0]  # Return first audio output (now always 24kHz WAV)
         return None
 
     demo = create_main_demo_ui(

pipeline_ace_step.py

@@ -1395,9 +1395,9 @@ class ACEStepPipeline:
         self,
         latents,
         target_wav_duration_second=30,
-        sample_rate=48000,
+        sample_rate=24000,
         save_path=None,
-        format="mp3",
+        format="wav",
     ):
         output_audio_paths = []
         bs = latents.shape[0]
@@ -1414,7 +1414,7 @@ class ACEStepPipeline:
         return output_audio_paths
 
     def save_wav_file(
-        self, target_wav, idx, save_path=None, sample_rate=48000, format="mp3"
+        self, target_wav, idx, save_path=None, sample_rate=24000, format="wav"
     ):
         if save_path is None:
             logger.warning("save_path is None, using default path ./outputs/")
@@ -1485,7 +1485,7 @@ class ACEStepPipeline:
         edit_n_max: float = 1.0,
         edit_n_avg: int = 1,
         save_path: str = None,
-        format: str = "mp3",
+        format: str = "wav",
         batch_size: int = 1,
         debug: bool = False,
     ):