python

scripts/deep_phonemizer/dp_export.py

@@ -75,7 +75,7 @@ def forward(self, text, phonemes=None, start_index=None):
 metadata = {
     "languages": "de en_us",
     "text_symbols": "a b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z ä ö ü Ä Ö Ü ß",
-    "phoneme_symbols": "a b d e f g h i j k l m n o p r s t u v w x y z æ ç ð ø ŋ œ ɐ ɑ ɔ ə ɛ ɜ ɹ ɡ ɪ ʁ ʃ ʊ ʌ ʏ ʒ ʔ ' ˌ ː ̃ ̍ ̥ ̩ ̯ ͡ θ . , : ; ? !",
+    "phoneme_symbols": "a b d e f g h i j k l m n o p r s t u v w x y z æ ç ð ø ŋ œ ɐ ɑ ɔ ə ɛ ɜ ɹ ɡ ɪ ʁ ʃ ʊ ʌ ʏ ʒ ʔ ' ˌ ː ̃ ̍ ̥ ̩ ̯ ͡ θ . , : ; ? ! \" ( ) -",
     "char_repeats": "3" if isinstance(model, ForwardTransformer) else "1",
     "lowercase": "1"
 }

wrappers/babylon.py

@@ -1,13 +1,17 @@
 import ctypes
 import os
 
+current_dir = os.path.dirname(os.path.abspath(__file__))
+
+symbols = " abdefghijklmnoprstuvwxyzæçðøŋœɐɑɔəɛɜɹɡɪʁʃʊʌʏʒʔ'ˌː ̃ ̍ ̥ ̩ ̯ ͡θ.,:;?!\"()-"
+
 # Load the shared library
 if os.name == 'nt':  # Windows
-    babylon_lib = ctypes.CDLL('./windows/libbabylon.dll')
+    babylon_lib = ctypes.CDLL(os.path.join(current_dir, 'windows', 'libbabylon.dll'))
 elif os.name == 'posix':  # macOS
-    babylon_lib = ctypes.CDLL('./macos/libbabylon.dylib')
+    babylon_lib = ctypes.CDLL(os.path.join(current_dir, 'macos', 'libbabylon.dylib'))
 else:  # Linux/Unix
-    babylon_lib = ctypes.CDLL('./linux/libbabylon.so')
+    babylon_lib = ctypes.CDLL(os.path.join(current_dir, 'linux', 'libbabylon.so'))
 
 # Define the function prototypes
 babylon_lib.babylon_g2p_init.argtypes = [ctypes.c_char_p, ctypes.c_char_p, ctypes.c_int]
@@ -71,8 +75,8 @@ def free_tts():
 
 # Example usage
 if __name__ == '__main__':
-    g2p_model_path = './models/deep_phonemizer.onnx'
-    tts_model_path = './models/curie.onnx'
+    g2p_model_path = os.path.join(current_dir, "models", "deep_phonemizer.onnx")
+    tts_model_path = os.path.join(current_dir, "models", "curie.onnx")
     language = 'en_us'
     use_punctuation = 1
     sequence = 'Hello world, This is a python test of babylon'