Spaces:
Running
Running
jhj0517
commited on
Commit
·
4a23819
1
Parent(s):
131f180
Handle None value
Browse files
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 |
-
|
|
|
|
|
|
|
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()
|