Update modules/whisper/whisper_base.py
Browse files
modules/whisper/whisper_base.py
CHANGED
@@ -216,11 +216,13 @@ class WhisperBase(ABC):
|
|
216 |
if files and isinstance(files[0], gr.utils.NamedString):
|
217 |
files = [file.name for file in files]
|
218 |
|
|
|
|
|
|
|
219 |
files_info = {}
|
220 |
for file in files:
|
221 |
|
222 |
## Detect language
|
223 |
-
model = whisper.load_model("base")
|
224 |
#params = WhisperParameters.as_value(*whisper_params)
|
225 |
#model = whisper.load_model(params.model_size)
|
226 |
mel = whisper.log_mel_spectrogram(whisper.pad_or_trim(whisper.load_audio(file))).to(model.device)
|
|
|
216 |
if files and isinstance(files[0], gr.utils.NamedString):
|
217 |
files = [file.name for file in files]
|
218 |
|
219 |
+
## Load model to detect language
|
220 |
+
model = whisper.load_model("base")
|
221 |
+
|
222 |
files_info = {}
|
223 |
for file in files:
|
224 |
|
225 |
## Detect language
|
|
|
226 |
#params = WhisperParameters.as_value(*whisper_params)
|
227 |
#model = whisper.load_model(params.model_size)
|
228 |
mel = whisper.log_mel_spectrogram(whisper.pad_or_trim(whisper.load_audio(file))).to(model.device)
|