Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -218,18 +218,11 @@ class WhisperBase(ABC):
 
             files_info = {}
             for file in files:
-                
-                transcribed_segments, time_for_task = self.run(
-                    file,
-                    progress,
-                    add_timestamp,
-                    *whisper_params,
-                )
 
                 ## Detect language
-                #model = whisper.load_model("base")
-                params = WhisperParameters.as_value(*whisper_params)
-                model = whisper.load_model(params.model_size)
+                model = whisper.load_model("base")
+                #params = WhisperParameters.as_value(*whisper_params)
+                #model = whisper.load_model(params.model_size)
                 mel = whisper.log_mel_spectrogram(whisper.pad_or_trim(whisper.load_audio(file))).to(model.device)
                 _, probs = model.detect_language(mel)
                 file_language = "not"
@@ -238,6 +231,13 @@ class WhisperBase(ABC):
                         file_language = value.capitalize()
                         break
                 
+                transcribed_segments, time_for_task = self.run(
+                    file,
+                    progress,
+                    add_timestamp,
+                    *whisper_params,
+                )
+                
                 file_name, file_ext = os.path.splitext(os.path.basename(file))
                 subtitle, file_path = self.generate_and_write_file(
                     file_name=file_name,