Update soni_translate/speech_segmentation.py

soni_translate/speech_segmentation.py

@@ -9,7 +9,6 @@ import gc
 import os
 import soundfile as sf
 from IPython.utils import capture # noqa
-from .language_configuration import EXTRA_ALIGN, INVERTED_LANGUAGES
 from .logging_setup import logger
 from .postprocessor import sanitize_file_name
 from .utils import remove_directory_contents, run_command
@@ -59,8 +58,6 @@ def load_align_and_align_segments(result, audio, DAMHF):
         language_code=result["language"],
         device=os.environ.get("SONITR_DEVICE") if os.environ.get("ZERO_GPU") != "TRUE" else "cuda",
         model_name=None
-        if result["language"] in DAMHF.keys()
-        else EXTRA_ALIGN[result["language"]],
     )
 
     # Align segments
@@ -343,31 +340,6 @@ def align_speech(audio, result):
     - Cleans up memory by releasing resources after alignment.
     """
     DAMHF.update(DAMT)  # lang align
-    if (
-        not result["language"] in DAMHF.keys()
-        and not result["language"] in EXTRA_ALIGN.keys()
-    ):
-        logger.warning(
-            "Automatic detection: Source language not compatible with align"
-        )
-        raise ValueError(
-            f"Detected language {result['language']}  incompatible, "
-            "you can select the source language to avoid this error."
-        )
-    if (
-        result["language"] in EXTRA_ALIGN.keys()
-        and EXTRA_ALIGN[result["language"]] == ""
-    ):
-        lang_name = (
-            INVERTED_LANGUAGES[result["language"]]
-            if result["language"] in INVERTED_LANGUAGES.keys()
-            else result["language"]
-        )
-        logger.warning(
-            "No compatible wav2vec2 model found "
-            f"for the language '{lang_name}', skipping alignment."
-        )
-        return result
 
     # random_sleep()
     result = load_align_and_align_segments(result, audio, DAMHF)