Spaces:
Runtime error
Runtime error
File size: 1,757 Bytes
b546670 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
# 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(" ")
st.caption("🗒️ This is a demo from our latest paper, **Nix-TTS**. <br> You can access the paper and the released models [here](https://github.com/rendchevi/nix-tts). <br> Authors: Rendi Chevi, Radityo Eko Prasojo, Alham Fikri Aji.<br> **Corresponding Author**: Rendi Chevi | rendi.chevi@{kata.ai, gmail.com}.", 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)
|