hzrr commited on
Commit
37332b5
·
1 Parent(s): e86df02

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import infer_tool
8
 
9
  convert_cnt = [0]
10
  dev = torch.device("cuda" if torch.cuda.is_available() else "cpu")
11
- model_name = "G.pth"
12
  config_name = "config.json"
13
  net_g_ms, hubert_soft, feature_input, hps_ms = infer_tool.load_model(f"{model_name}", f"configs/{config_name}")
14
 
 
8
 
9
  convert_cnt = [0]
10
  dev = torch.device("cuda" if torch.cuda.is_available() else "cpu")
11
+ model_name = "1121_epochs.pth"
12
  config_name = "config.json"
13
  net_g_ms, hubert_soft, feature_input, hps_ms = infer_tool.load_model(f"{model_name}", f"configs/{config_name}")
14