autoregressive

app.py

@@ -4,7 +4,9 @@ from gruut import sentences
 from phonemap.g2pphonemizer import G2PPhonemizer
 from phonemap.charsiu import CharsiuPhonemizer
 from openphonemizer import OpenPhonemizer
+from cached_path import cached_path
 opphonemizer = OpenPhonemizer()
+opphonemizer_ar = OpenPhonemizer(str(cached_path('hf://openphonemizer/autoreg-ckpt/best_model.pt')))
 
 g2p = G2PPhonemizer()
 charsiu = CharsiuPhonemizer()
@@ -12,7 +14,6 @@ import phonemizer
 global_phonemizer = phonemizer.backend.EspeakBackend(language='en-us', preserve_punctuation=True, with_stress=True, words_mismatch='ignore')
 from nltk import word_tokenize
 import nltk
-from cached_path import cached_path
 
 from dp.phonemizer import Phonemizer
 phonemizer = Phonemizer.from_checkpoint(str(cached_path('https://public-asai-dl-models.s3.eu-central-1.amazonaws.com/DeepPhonemizer/en_us_cmudict_ipa_forward.pt')))
@@ -55,17 +56,20 @@ def dp(text):
     return phonemizer(text, lang='en_us')
 def run_openphonemizer(text):
     return opphonemizer(text)
+def opar(text):
+    return opphonemizer_ar(text)
 def run(t, r):
     if r == 'phonemizer': return phonemizerfunc(t)
     if r == 'openphonemizer': return run_openphonemizer(t)
     if r == 'gruut': return gruut(t)
     if r == 'deep_phonemizer': return dp(t)
     if r == 'g2p_en (buggy)': return g2pen(t)
+    if r == 'openphonemizer autoregressive': return opar(t)
     return f'Error: Phonemizer {r} not found, please try another Phonemizer and create a Community discussion on HF.'
 with gr.Blocks() as demo:
     gr.Markdown("# PhonemizerHub")
     t = gr.Textbox(label="Text", placeholder="Enter text...", interactive=True)
-    r = gr.Radio(choices=['phonemizer', 'openphonemizer', 'gruut', 'deep_phonemizer', 'g2p_en (buggy)'], value='openphonemizer', interactive=True, label='Select a phonemizer')
+    r = gr.Radio(choices=['phonemizer', 'openphonemizer', 'gruut', 'openphonemizer autoregressive', 'deep_phonemizer', 'g2p_en (buggy)'], value='openphonemizer', interactive=True, label='Select a phonemizer')
     b = gr.Button("Phonemize")
     out = gr.Textbox(interactive=False, label="Phonemes")
     b.click(run, inputs=[t, r], outputs=[out])