Make dictionary non-necessary for character input

xd009642 · Apr 22, 2024 · b467d0c · b467d0c
1 parent f3c8b8c
commit b467d0c
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -38,10 +38,15 @@ pub struct XdTts {
 
 impl XdTts {
     pub fn new(tacotron2: &Path, phoneme_input: bool) -> anyhow::Result<Self> {
-        let mut dict = CmuDictionary::open("data/cmudict-0.7b.txt")?;
-        if let Ok(custom) = CmuDictionary::open("resources/custom_dict.txt") {
-            dict.merge(custom);
-        }
+        let dict = if phoneme_input {
+            let mut dict = CmuDictionary::open("data/cmudict-0.7b.txt")?;
+            if let Ok(custom) = CmuDictionary::open("resources/custom_dict.txt") {
+                dict.merge(custom);
+            }
+            dict
+        } else {
+            CmuDictionary::default()
+        };
         let model = Tacotron2::load(tacotron2)?;
         let vocoder = create_griffin_lim()?;
         Ok(Self {