fix asr import

abdeladim-s · Jun 2, 2023 · 4bb07cc · 4bb07cc
1 parent 3006094
commit 4bb07cc
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 2 deletions.
diff --git a/easymms/models/alignment.py b/easymms/models/alignment.py
@@ -11,6 +11,17 @@
 from pathlib import Path
 from typing import List
 import torch
+# fix importing from fairseq.examples
+# import site
+# sys.path.append(str(Path(site.getsitepackages()[0]) / 'fairseq'))
+# try:
+#     from fairseq.examples.mms.data_prep.align_and_segment import get_alignments
+#     from fairseq.examples.mms.data_prep.align_utils import get_uroman_tokens, get_spans
+#     from fairseq.examples.mms.data_prep.text_normalization import text_normalize
+# except ImportError:
+#     from examples.mms.data_prep.align_and_segment import get_alignments
+#     from examples.mms.data_prep.align_utils import get_uroman_tokens, get_spans
+#     from examples.mms.data_prep.text_normalization import text_normalize
 
 from easymms import utils
 from easymms._logger import set_log_level

diff --git a/easymms/models/asr.py b/easymms/models/asr.py
@@ -23,6 +23,14 @@
 from pydub import AudioSegment
 from pydub.utils import mediainfo
 
+# fix importing from fairseq.examples
+# import site
+# sys.path.append(str(Path(site.getsitepackages()[0]) / 'fairseq'))
+# try:
+#     from fairseq.examples.speech_recognition.new.infer import hydra_main
+# except ImportError:
+#     from examples.speech_recognition.new.infer import hydra_main
+
 
 from easymms import utils as easymms_utils
 from easymms._logger import set_log_level
@@ -142,12 +150,16 @@ def transcribe(self,
 
         :return: List of transcription text in the same order as input files
         """
+        processed_files = self._prepare_media_files(media_files)
         # import
         cwd = os.getcwd()
         os.chdir(constants.FAIRSEQ_DIR)
         from examples.speech_recognition.new.infer import hydra_main
+        import site
+        fairseq_path = str((Path(site.getsitepackages()[0]) / 'fairseq').resolve())
+        sys.path.append(fairseq_path)
+        os.chdir(fairseq_path)
 
-        processed_files = self._prepare_media_files(media_files)
         self._setup_tmp_dir(processed_files)
         # edit cfg
         if cfg is None:
@@ -186,8 +198,8 @@ def transcribe(self,
                 res.append(segments)
         else:
             res = transcripts
-
         os.chdir(cwd)
+
         return res
 
     @staticmethod