Spaces:
Build error
Build error
...
Browse files
app.py
CHANGED
@@ -45,8 +45,8 @@ def load_model(config_path, pth_path):
|
|
45 |
|
46 |
print(f"{pth_path}加载成功!")
|
47 |
|
48 |
-
def infer(
|
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)
|