Update audio_foundation_models.py

audio_foundation_models.py

@@ -978,10 +978,8 @@ class Speech_Enh_SC:
                          "representing the audio_path. ")
     def inference(self, speech_path, ref_channel=0):
         speech, sr = soundfile.read(speech_path)
-        print("###")
-        print(speech)
-        # speech = speech[:, ref_channel]
-        # print(speech)
+        if speech.ndim != 1:
+            speech = speech[:, ref_channel]
         enh_speech = self.separate_speech(speech[None, ...], fs=sr)
         audio_filename = os.path.join('audio', str(uuid.uuid4())[0:8] + ".wav")
         soundfile.write(audio_filename, enh_speech[0].squeeze(), samplerate=sr)