Update app.py

app.py

@@ -35,6 +35,7 @@ class TTS:
         self.voicegenerator.load_state_dict(check_point['gen_model'])
         self.voicegenerator.eval()
         self.voicegenerator.remove_weight_norm()
+        self.finish = True
     
     def inference(self, input_text, noise_scale = 0.667, length_scale = 1.0):
         filters = '([.,!?])'
@@ -97,11 +98,12 @@ with col2:
 button_change = st.button("Change Vocie") 
 if button_change == True:
     if model_variant != st.session_state.model_variant:
-    # Update variant choice
-    update_session_state("model_variant", model_variant)
-    # Re-load model
-    update_model()
-    st.snow()
+        # Update variant choice
+        update_session_state("model_variant", model_variant)
+        # Re-load model
+        with st.spinner('Speaker updating'):
+            update_model()
+            st.success('Done!', icon="✅")
     
 noise_scale = st.slider('noise를 추가합니다.', 0., 2., value = 0.33, step = 0.01)
 length_scale = st.slider('속도를 조절합니다.', 0., 2., value = 1., step = 0.01)