Update app.py

app.py

@@ -5,6 +5,7 @@ import torch
 from datautils import *
 from model import Generator as Glow_model
 from Hmodel import Generator as GAN_model
+from Tmodel import GlowTTS as T_Glow_model
 
 st.set_page_config(
     page_title = "소신 Team Demo",
@@ -16,7 +17,7 @@ class TTS:
         global device
         device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         torch.cuda.manual_seed(1234) if torch.cuda.is_available() else None
-        self.flowgenerator = Glow_model(n_vocab = 70 if model_variant != '태연' else 73, h_c= 192, f_c = 768, f_c_dp = 256, out_c = 80, k_s = 3, k_s_dec = 5, heads=2, layers_enc = 6).to(device)
+        self.flowgenerator = Glow_model(n_vocab = 70 , h_c= 192, f_c = 768, f_c_dp = 256, out_c = 80, k_s = 3, k_s_dec = 5, heads=2, layers_enc = 6).to(device) if model_variant != '태연' \else T_Glow_model()
         self.voicegenerator = GAN_model().to(device)
         if model_variant == '감기걸린 은식':
             name = '1038_eunsik_01'