LAP-DEV commited on
Commit
327d7eb
·
verified ·
1 Parent(s): 4ad9ce5

Update modules/whisper/whisper_base.py

Browse files
Files changed (1) hide show
  1. modules/whisper/whisper_base.py +3 -1
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)