From 9f8982181d9f7159b18441fe3740f29879bd4528 Mon Sep 17 00:00:00 2001
From: Plachta <liusongting07@gmail.com>
Date: Wed, 1 Mar 2023 17:11:58 +0800
Subject: [PATCH] updated pure chinese option

---
 long_audio_transcribe.py  | 4 ++++
 short_audio_transcribe.py | 4 ++++
 2 files changed, 8 insertions(+)

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]