aadnk commited on
Commit
e77e696
1 Parent(s): e694402

Fix chinese

Browse files
src/whisper/fasterWhisperContainer.py CHANGED
@@ -180,7 +180,7 @@ class FasterWhisperCallback(AbstractWhisperCallback):
180
 
181
  def _lookup_language_code(self, language: str):
182
  lookup = {
183
- "english": "en", "chinese": "zh-cn", "german": "de", "spanish": "es", "russian": "ru", "korean": "ko",
184
  "french": "fr", "japanese": "ja", "portuguese": "pt", "turkish": "tr", "polish": "pl", "catalan": "ca",
185
  "dutch": "nl", "arabic": "ar", "swedish": "sv", "italian": "it", "indonesian": "id", "hindi": "hi",
186
  "finnish": "fi", "vietnamese": "vi", "hebrew": "he", "ukrainian": "uk", "greek": "el", "malay": "ms",
 
180
 
181
  def _lookup_language_code(self, language: str):
182
  lookup = {
183
+ "english": "en", "chinese": "zh", "german": "de", "spanish": "es", "russian": "ru", "korean": "ko",
184
  "french": "fr", "japanese": "ja", "portuguese": "pt", "turkish": "tr", "polish": "pl", "catalan": "ca",
185
  "dutch": "nl", "arabic": "ar", "swedish": "sv", "italian": "it", "indonesian": "id", "hindi": "hi",
186
  "finnish": "fi", "vietnamese": "vi", "hebrew": "he", "ukrainian": "uk", "greek": "el", "malay": "ms",