Spaces:
Runtime error
Runtime error
| # Utils | |
| import os | |
| import soundfile as sf | |
| # Streamlit | |
| import streamlit as st | |
| # Custom elements | |
| from elements.component import ( | |
| centered_text, | |
| ) | |
| from elements.session_states import ( | |
| init_session_state, | |
| update_session_state, | |
| update_model, | |
| ) | |
| from elements.tts import ( | |
| generate_voice, | |
| ) | |
| st.set_page_config( | |
| page_title = "Nix-TTS Interactive Demo", | |
| page_icon = "🐤", | |
| ) | |
| # Initiate stuffs | |
| init_session_state() | |
| # --------------------------------------------------------------------------------- | |
| # Description | |
| centered_text("🐤 Nix-TTS Interactive Demo") | |
| centered_text("An incredibly lightweight end-to-end text-to-speech model via knowledge distillation", "h5") | |
| st.write(" ") | |
| mode = "p" | |
| st.markdown( | |
| f"<{mode} style='text-align: left;'><small>🗒️ This is a demo from our latest paper, <b>Nix-TTS</b> (Submitted to <b>INTERSPEECH 2022</b>)<br> You can access the paper and the released models <a href='https://github.com/rendchevi/nix-tts'>here</a><br> Authors: Rendi Chevi, Radityo Eko Prasojo, Alham Fikri Aji.<br> Corresponding Author: Rendi Chevi, rendi.chevi@{kata.ai, gmail.com}</small></{mode}>", | |
| unsafe_allow_html = True | |
| ) | |
| # Model demo | |
| st.write(" ") | |
| st.write(" ") | |
| col1, col2 = st.columns(2) | |
| with col1: | |
| input_text = st.text_input( | |
| "Input Text", | |
| value = "Born to multiply, born to gaze into night skies.", | |
| ) | |
| with col2: | |
| model_variant = st.selectbox("Choose Model Variant", options = ["Deterministic", "Stochastic"], index = 1) | |
| if model_variant != st.session_state.model_variant: | |
| # Update variant choice | |
| update_session_state("model_variant", model_variant) | |
| # Re-load model | |
| update_model() | |
| button_gen = st.button("Generate Voice") | |
| if button_gen == True: | |
| generate_voice(input_text) | |