marigold334 commited on
Commit
8ebd890
โ€ข
1 Parent(s): 1a91ed2
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -5,6 +5,7 @@ import torch
5
  from datautils import *
6
  from model import Generator as Glow_model
7
  from Hmodel import Generator as GAN_model
 
8
 
9
  st.set_page_config(
10
  page_title = "์†Œ์‹  Team Demo",
@@ -16,7 +17,7 @@ class TTS:
16
  global device
17
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
18
  torch.cuda.manual_seed(1234) if torch.cuda.is_available() else None
19
- 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)
20
  self.voicegenerator = GAN_model().to(device)
21
  if model_variant == '๊ฐ๊ธฐ๊ฑธ๋ฆฐ ์€์‹':
22
  name = '1038_eunsik_01'
 
5
  from datautils import *
6
  from model import Generator as Glow_model
7
  from Hmodel import Generator as GAN_model
8
+ from Tmodel import GlowTTS as T_Glow_model
9
 
10
  st.set_page_config(
11
  page_title = "์†Œ์‹  Team Demo",
 
17
  global device
18
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
19
  torch.cuda.manual_seed(1234) if torch.cuda.is_available() else None
20
+ 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()
21
  self.voicegenerator = GAN_model().to(device)
22
  if model_variant == '๊ฐ๊ธฐ๊ฑธ๋ฆฐ ์€์‹':
23
  name = '1038_eunsik_01'