jhj0517 commited on
Commit
4a23819
·
1 Parent(s): 131f180

Handle None value

Browse files
Files changed (1) hide show
  1. modules/whisper/whisper_base.py +4 -1
modules/whisper/whisper_base.py CHANGED
@@ -122,7 +122,10 @@ class WhisperBase(ABC):
122
 
123
  if audio.ndim >= 2:
124
  audio = audio.mean(axis=1)
125
- origin_sample_rate = self.music_separator.audio_info.sample_rate
 
 
 
126
  audio = self.resample_audio(audio=audio, original_sample_rate=origin_sample_rate)
127
 
128
  self.music_separator.offload()
 
122
 
123
  if audio.ndim >= 2:
124
  audio = audio.mean(axis=1)
125
+ if self.music_separator.audio_info is not None:
126
+ origin_sample_rate = self.music_separator.audio_info.sample_rate
127
+ else:
128
+ origin_sample_rate = 16000
129
  audio = self.resample_audio(audio=audio, original_sample_rate=origin_sample_rate)
130
 
131
  self.music_separator.offload()