fix transliteration

GooeyAI · Jun 26, 2024 · 7ceef32 · 7ceef32
1 parent c7a7136
commit 7ceef32
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/daras_ai_v2/asr.py b/daras_ai_v2/asr.py
@@ -565,8 +565,7 @@ def run_google_translate(
     if not target_language:
         raise UserError(f"Unsupported target language: {target_language!r}")
 
-    # if the language supports transliteration, we should check if the script is Latin
-    if source_language and source_language not in TRANSLITERATION_SUPPORTED:
+    if source_language:
         language_codes = [source_language] * len(texts)
     else:
         translate_client = translate.Client()
@@ -592,11 +591,8 @@ def _translate_text(
     target_language: str,
     glossary_url: str | None,
 ) -> str:
-    is_romanized = source_language.endswith("-Latn")
     source_language = source_language.replace("-Latn", "")
-    enable_transliteration = (
-        is_romanized and source_language in TRANSLITERATION_SUPPORTED
-    )
+    enable_transliteration = source_language in TRANSLITERATION_SUPPORTED
 
     # prevent incorrect API calls
     if not text or source_language == target_language or source_language == "und":