Update app.py

app.py

@@ -21,13 +21,16 @@ class TTS:
         if model_variant == '은식':
             name = '1038_eunsik_01'
             last_chpt1 = './log/1038_eunsik_01/Glow_TTS_00289602.pt'
+        elif model_variant == 'KSS':
+            last_chpt1 = './log/KSS/Glow_TTS_00280641.pt'
         check_point = torch.load(last_chpt1, map_location = device)
         self.flowgenerator.load_state_dict(check_point['generator'])
         self.flowgenerator.decoder.skip()
         self.flowgenerator.eval()
         if model_variant == '은식':
-            name = '1038_eunsik_01'
-            last_chpt2 = './log/1038_eunsik_01/HiFI_GAN_00257000.pt'
+            last_chpt2 = './log/1038_eunsik_01/HiFI_GAN_00257000.pt'\
+        elif model_variant == 'KSS':
+            last_chpt1 = './log/KSS/HiFi_GAN_00135000.pt'
         check_point = torch.load(last_chpt2, map_location = device)
         self.voicegenerator.load_state_dict(check_point['gen_model'])
         self.voicegenerator.eval()
@@ -89,7 +92,7 @@ st.write(" ")
 
 mode = "p"
 st.markdown(
-    f"<{mode} style='text-align: left;'><small>This is a demo trained by our vocie.&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp; The voice \"KSS\" is traind 3 times \"은식\" is finetuned from \"KSS\" for 3 times &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp; We got this deomoformat from Nix-TTS Interactive Demo</small></{mode}>",
+    f"<{mode} style='text-align: left;'><small>This is a demo trained by our vocie. The voice \"KSS\" is traind by KSS Dataset. \"은식\" which is about 1 hour audio is finetuned from \"KSS\". We got this deomoformat from Nix-TTS Interactive Demo</small></{mode}>",
     unsafe_allow_html = True
 )
 
@@ -113,5 +116,6 @@ with col2:
 button_gen = st.button("Generate Voice")
 if button_gen == True:
     generate_voice(input_text)
+    st.balloons()