Update app.py

app.py

@@ -28,7 +28,7 @@ class TTS:
             last_chpt1 = './log/Taeyeon/Glow_TTS_337000.pt'
         check_point = torch.load(last_chpt1, map_location = device)
         self.flowgenerator.load_state_dict(check_point['generator' if model_variant != '태연' else 'model'])
-        self.flowgenerator.decoder.skip()
+        self.flowgenerator.decoder.skip() if model_variant != '태연' else None
         self.flowgenerator.eval()
         if model_variant == '감기걸린 은식':
             last_chpt2 = './log/1038_eunsik_01/HiFI_GAN_00257000.pt'
@@ -49,7 +49,7 @@ class TTS:
         x_length = torch.tensor(x.shape[1]).unsqueeze(0).to(device)
         
         with torch.no_grad():
-            (y_gen_tst, *_), *_, (attn_gen, *_) = self.flowgenerator(x, x_length, gen = True, noise_scale = noise_scale, length_scale = length_scale)
+                (y_gen_tst, *_), *_, (attn_gen, *_) = self.flowgenerator(x, x_length, gen = True, noise_scale = noise_scale, length_scale = length_scale) if st.session_state.model_variant != "태연" else self.flowgenerator(x, x_length, inference = True, noise_scale = noise_scale, length_scale = length_scale) 
             y = self.voicegenerator(y_gen_tst)
             audio = y.squeeze() * 32768.0
             voice = audio.cpu().numpy().astype('int16')