Update app.py

app.py

@@ -641,8 +641,9 @@ pro=TrinerModelVITS(dir_model=dir_model,
                                                             token=token,
                                                                      device=device
                     )
-def loadd_d():
+def loadd_d(n_model):
     token=os.environ.get("key_")
+    model=VitsModel.from_pretrained(n_model,token=token)
     return token
 @spaces.GPU(duration=30)
 def run_train_epoch(num):
@@ -688,7 +689,7 @@ with gr.Blocks() as interface:
         output_i  = gr.Textbox(label="namemodel")
         output_ini = gr.Textbox(label="token")
         label=gr.Label("hhh")
-        btn_init.click(loadd_d,[],[label])
+        btn_init.click(loadd_d,[output_i],[label])
     with gr.Accordion("init_Starting ", open=False):
         btn_init = gr.Button("init start")
         output_init = gr.Textbox(label="init")