Spaces:
Build error
Build error
...
Browse files
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)
|