Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -228,7 +228,7 @@ class WhisperBase(ABC):
                 #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"
+                file_language = ""
                 for key,value in whisper.tokenizer.LANGUAGES.items():
                     if key == str(max(probs, key=probs.get)):
                         file_language = value.capitalize()
@@ -281,10 +281,10 @@ class WhisperBase(ABC):
                 total_result += f'{info["subtitle"]}'
                 total_time += info["time_for_task"]
                 #total_info += f'{info["lang"]}'
-                total_info += f"Language {info['lang']} detected"
+                total_info += f"Processed file: {file}\nLanguage prediction: {info['lang']}"
             
             #result_str = f"Processing of file '{file_name}{file_ext}' done in {self.format_time(total_time)}:\n\n{total_result}"
-            total_info += f"\nTranscription process done in {self.format_time(total_time)}"
+            total_info += f"\nTranscription duration: {self.format_time(total_time)}"
             result_str = total_result
             result_file_path = [info['path'] for info in files_to_download.values()]