hzrr commited on
Commit
0281d5f
·
1 Parent(s): 3bf0ca5
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -45,8 +45,8 @@ def load_model(config_path, pth_path):
45
 
46
  print(f"{pth_path}加载成功!")
47
 
48
- def infer(c_id, text):
49
-
50
  stn_tst = get_text(text, hps)
51
  with torch.no_grad():
52
  x_tst = stn_tst.to(dev).unsqueeze(0)
 
45
 
46
  print(f"{pth_path}加载成功!")
47
 
48
+ def infer(c_name, text):
49
+ c_id = character_dict[c_name]
50
  stn_tst = get_text(text, hps)
51
  with torch.no_grad():
52
  x_tst = stn_tst.to(dev).unsqueeze(0)