Update app.py

app.py

@@ -6,13 +6,18 @@ from datautils import *
 from model import Generator as Glow_model
 from Hmodel import Generator as GAN_model
 
-device = torch.device('cuda:0') if torch.cuda.is_available() else 'cpu'
-torch.cuda.manual_seed(1234) if torch.cuda.is_available() else None
+st.set_page_config(
+    page_title = "소신 Team Demo",
+    page_icon = "🔉",
+)
 
 class TTS:
     def __init__(self, model_variant):
-        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)
-        self.voicegenerator = GAN_model()
+        global device
+        device = torch.device('cuda:0') 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, 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.voicegenerator = GAN_model().to(device)
         if model_variant == '은식':
             name = '1038_eunsik_01'
             last_chpt1 = './log/1038_eunsik_01/Glow_TTS_00289602.pt'
@@ -77,10 +82,7 @@ def generate_voice(input_text):
     st.caption("Generated Voice")
 
 
-st.set_page_config(
-    page_title = "소신 Team Demo",
-    page_icon = "🔉",
-)
+
 
 init_session_state()