handler change0

handler.py

@@ -27,12 +27,13 @@ class EndpointHandler:
         
         self.model.set_generation_params(duration=duration)
         outputs = self.model.generate(inputs)
-        prediction = outputs[0].cpu().numpy()
 
-        # Define the output file path
-        output_file_path = "generated_audio.wav"
+        # Save each generated audio file with loudness normalization
+        output_files = []
+        for idx, one_wav in enumerate(outputs):
+            output_file_path = f"generated_audio_{idx}.wav"
+            audio_write(output_file_path, one_wav.cpu(), self.model.sample_rate, strategy="loudness", loudness_compressor=True)
+            output_files.append(output_file_path)
         
-        # Save the generated audio to a file
-        audio_write(output_file_path, prediction, sample_rate=16000)  # Adjust sample_rate as needed
-
-        return {"generated_audio_file": output_file_path}
+        return {"generated_audio_files": output_files}
+