Update app.py

app.py

@@ -36,7 +36,7 @@ class TTS:
         self.voicegenerator.eval()
         self.voicegenerator.remove_weight_norm()
     
-    def inference(self, input_textm, noise_scale = 0.667, length_scale = 1.0):
+    def inference(self, input_text, noise_scale = 0.667, length_scale = 1.0):
         filters = '([.,!?])'
         sentence = re.sub(re.compile(filters), '', input_text)
         x = text_to_sequence(sentence)
@@ -70,15 +70,6 @@ def centered_text(input_text, mode = "h1",):
     st.markdown(
         f"<{mode} style='text-align: center;'>{input_text}</{mode}>", unsafe_allow_html = True)
 
-def generate_voice(input_text):
-    # TTS Inference
-    voice = st.session_state.TTS.inference(input_text)
-    
-    # Play audio
-    st.audio(voice,sample_rate = 22050)
-    st.caption("Generated Voice by" + st.session_state.model_variant)
-
-
 init_session_state()
 
 centered_text("🔉 소신 Team Demo")
@@ -109,11 +100,13 @@ with col2:
         update_model()
         st.snow()
         
-noise_scale = st.slider('noise를 추가합니다.', 0., 2., value = 0.66, step = 0.01)
+noise_scale = st.slider('noise를 추가합니다.', 0., 2., value = 0.33, step = 0.01)
 length_scale = st.slider('속도를 조절합니다.', 0., 2., value = 1., step = 0.01)
 button_gen = st.button("Generate Voice")
 if button_gen == True:
-    generate_voice(input_text, noise_scale, length_scale)
+    voice = st.session_state.TTS.inference(input_text, noise_scale, length_scale)
+    st.audio(voice,sample_rate = 22050)
+    st.caption("Generated Voice by" + st.session_state.model_variant)
     st.balloons()