Update modules/whisper/whisper_base.py
Browse files
    	
        modules/whisper/whisper_base.py
    CHANGED
    
    | @@ -270,6 +270,7 @@ class WhisperBase(ABC): | |
| 270 | 
             
                            #files_info[file_name] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language, "lang_prob": file_lang_probs, "input_source_file": (file_name+file_ext)}
         | 
| 271 |  | 
| 272 | 
             
                            # Translate the transcribed segments
         | 
|  | |
| 273 | 
             
                            if translate_output:
         | 
| 274 | 
             
                                self.nllb_inf = NLLBInference()
         | 
| 275 | 
             
                                if file_language in NLLB_AVAILABLE_LANGS.keys():
         | 
| @@ -280,7 +281,7 @@ class WhisperBase(ABC): | |
| 280 | 
             
                                        tgt_lang=target_lang,
         | 
| 281 | 
             
                                        speaker_diarization=params.is_diarize
         | 
| 282 | 
             
                                    )
         | 
| 283 | 
            -
                                    translation_note =  | 
| 284 | 
             
                                else:
         | 
| 285 | 
             
                                    translation_note = file_language + " not supported"
         | 
| 286 |  | 
|  | |
| 270 | 
             
                            #files_info[file_name] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language, "lang_prob": file_lang_probs, "input_source_file": (file_name+file_ext)}
         | 
| 271 |  | 
| 272 | 
             
                            # Translate the transcribed segments
         | 
| 273 | 
            +
                            translation_note = ""
         | 
| 274 | 
             
                            if translate_output:
         | 
| 275 | 
             
                                self.nllb_inf = NLLBInference()
         | 
| 276 | 
             
                                if file_language in NLLB_AVAILABLE_LANGS.keys():
         | 
|  | |
| 281 | 
             
                                        tgt_lang=target_lang,
         | 
| 282 | 
             
                                        speaker_diarization=params.is_diarize
         | 
| 283 | 
             
                                    )
         | 
| 284 | 
            +
                                    translation_note = file_language + " to " + target_lang
         | 
| 285 | 
             
                                else:
         | 
| 286 | 
             
                                    translation_note = file_language + " not supported"
         | 
| 287 |  |