Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -219,25 +219,24 @@ class WhisperBase(ABC):
             files_info = {}
             for file in files:
                 
-                ## Detect language
-                #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"
-#                for key,value in whisper.tokenizer.LANGUAGES.items():
-#                    if key == str(max(probs, key=probs.get)):
-#                        file_language = value.capitalize()
-#                        break	
-                file_language = "#ToDo#"
-                
                 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)
+                mel = whisper.log_mel_spectrogram(whisper.pad_or_trim(whisper.load_audio(file))).to(model.device)
+                _, probs = model.detect_language(mel)
+                file_language = "not"
+                for key,value in whisper.tokenizer.LANGUAGES.items():
+                    if key == str(max(probs, key=probs.get)):
+                        file_language = value.capitalize()
+                        break
                 
                 file_name, file_ext = os.path.splitext(os.path.basename(file))
                 subtitle, file_path = self.generate_and_write_file(