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