update test

test.py

@@ -147,33 +147,13 @@ def get_text(text, lexicon, tokens, punctuation):
     return ans
 
 
-def main():
-    model = OnnxModel("./vits-ljs.onnx")
-
-    lexicon = read_lexicon()
-    tokens = read_tokens()
-    convert_lexicon(lexicon, tokens)
-
-    x = get_text(
-        "Liliana, our most beautiful and lovely assistant",
-        lexicon,
-        tokens,
-        model.punctuation,
-    )
-    #  x = get_text(
-    #      "Ask not what your country can do for you; ask what you can do for your country.",
-    #      lexicon,
-    #      tokens,
-    #      model.punctuation,
-    #  )
-
+def generate(model, text, lexicon, tokens):
     x = get_text(
-        "Success is not final, failure is not fatal, it is the courage to continue that counts!",
+        text,
         lexicon,
         tokens,
         model.punctuation,
     )
-
     if model.add_blank:
         x2 = [0] * (2 * len(x) + 1)
         x2[1::2] = x
@@ -182,7 +162,28 @@ def main():
     x = torch.tensor(x, dtype=torch.int64)
 
     y = model(x)
-    soundfile.write("test.wav", y.numpy(), model.sample_rate)
+
+    return y
+
+
+def main():
+    model = OnnxModel("./vits-ljs.onnx")
+
+    lexicon = read_lexicon()
+    tokens = read_tokens()
+    convert_lexicon(lexicon, tokens)
+
+    text = "Liliana, our most beautiful and lovely assistant"
+    y = generate(model, text, lexicon, tokens)
+    soundfile.write("test-0.wav", y.numpy(), model.sample_rate)
+
+    text = "Ask not what your country can do for you; ask what you can do for your country."
+    y = generate(model, text, lexicon, tokens)
+    soundfile.write("test-1.wav", y.numpy(), model.sample_rate)
+
+    text = "Success is not final, failure is not fatal, it is the courage to continue that counts!"
+    y = generate(model, text, lexicon, tokens)
+    soundfile.write("test-2.wav", y.numpy(), model.sample_rate)
 
 
 if __name__ == "__main__":