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()
|