Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -220,15 +220,16 @@ class WhisperBase(ABC):
                 
                 ## 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)
+#                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 = "not"
-                for key,value in whisper.tokenizer.LANGUAGES.items():
-                    if key == str(max(probs, key=probs.get)):
-                        file_language = value.capitalize()
-                        break	
                 
                 transcribed_segments, time_for_task = self.run(
                     file,