Update networks.py

networks.py

@@ -239,7 +239,7 @@ class SpeechModel:
                 for idx in tqdm(range(num_samples)):  # Loop over all audio samples
                     self.data = {}
                     # Read the audio, waveform ID, and audio length from the data reader
-                    input_audio, wav_id, input_len = data_reader[idx]
+                    input_audio, wav_id, input_len, scalar = data_reader[idx]
                     # Store the input audio and metadata in self.data
                     self.data['audio'] = input_audio
                     self.data['id'] = wav_id
@@ -247,6 +247,12 @@ class SpeechModel:
                     
                     # Perform the audio decoding/processing
                     output_audio = self.decode()
+
+                    if isinstance(output_audio, list):
+                        for spk in range(self.args.num_spks):
+                            output_audio[spk] = output_audio[spk] * scalar
+                    else:
+                        output_audio = output_audio * scalar
                     
                     if online_write:
                         # If online writing is enabled, save the output audio to files