diff --git a/long_audio_transcribe.py b/long_audio_transcribe.py index af89147..92b2731 100644 --- a/long_audio_transcribe.py +++ b/long_audio_transcribe.py @@ -23,6 +23,10 @@ if __name__ == "__main__": 'zh': "[ZH]", 'ja': "[JA]", } + elif args.languages == "C": + lang2token = { + 'zh': "[ZH]", + } model = whisper.load_model(args.whisper_size) speaker_annos = [] for file in filelist: diff --git a/short_audio_transcribe.py b/short_audio_transcribe.py index f687db9..9cd851d 100644 --- a/short_audio_transcribe.py +++ b/short_audio_transcribe.py @@ -43,6 +43,10 @@ if __name__ == "__main__": 'zh': "[ZH]", 'ja': "[JA]", } + elif args.languages == "C": + lang2token = { + 'zh': "[ZH]", + } model = whisper.load_model(args.whisper_size) parent_dir = "./custom_character_voice/" speaker_names = list(os.walk(parent_dir))[0][1]