hzrr commited on
Commit
46d300f
·
1 Parent(s): 9d44aa3
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -48,6 +48,7 @@ def load_model(config_path, pth_path):
48
  def infer(c_id, text):
49
  if c_id not in list(range[1, 14]):
50
  raise gr.Error("角色id超出范围!")
 
51
  stn_tst = get_text(text, hps)
52
  with torch.no_grad():
53
  x_tst = stn_tst.to(dev).unsqueeze(0)
 
48
  def infer(c_id, text):
49
  if c_id not in list(range[1, 14]):
50
  raise gr.Error("角色id超出范围!")
51
+ print(c_id)
52
  stn_tst = get_text(text, hps)
53
  with torch.no_grad():
54
  x_tst = stn_tst.to(dev).unsqueeze(0)