修复app.py代码错误
Browse files
app.py
CHANGED
|
@@ -186,10 +186,14 @@ if __name__ == '__main__':
|
|
| 186 |
utils.load_checkpoint(model_path, model, None)
|
| 187 |
model.eval().to(device)
|
| 188 |
|
|
|
|
|
|
|
| 189 |
t = info["type"]
|
| 190 |
|
| 191 |
if t == "vits-fast":
|
|
|
|
| 192 |
for name, id in hps.speakers.items():
|
|
|
|
| 193 |
if name != "None":
|
| 194 |
speaker_ids.append(id)
|
| 195 |
speakers.append(name)
|
|
@@ -197,6 +201,11 @@ if __name__ == '__main__':
|
|
| 197 |
speaker_ids = [sid for sid, name in enumerate(hps.speakers) if name != "None"]
|
| 198 |
speakers = [name for sid, name in enumerate(hps.speakers) if name != "None"]
|
| 199 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 200 |
if t == "vits" or t == "vits-fast":
|
| 201 |
models_tts.append((name, author, cover_path, speakers, lang, example,
|
| 202 |
hps.symbols, create_tts_fn(model, hps, speaker_ids),
|
|
|
|
| 186 |
utils.load_checkpoint(model_path, model, None)
|
| 187 |
model.eval().to(device)
|
| 188 |
|
| 189 |
+
speaker_ids = []
|
| 190 |
+
speakers = []
|
| 191 |
t = info["type"]
|
| 192 |
|
| 193 |
if t == "vits-fast":
|
| 194 |
+
# print("vits-fast")
|
| 195 |
for name, id in hps.speakers.items():
|
| 196 |
+
# print(hps.speakers)
|
| 197 |
if name != "None":
|
| 198 |
speaker_ids.append(id)
|
| 199 |
speakers.append(name)
|
|
|
|
| 201 |
speaker_ids = [sid for sid, name in enumerate(hps.speakers) if name != "None"]
|
| 202 |
speakers = [name for sid, name in enumerate(hps.speakers) if name != "None"]
|
| 203 |
|
| 204 |
+
print("speaker_ids:")
|
| 205 |
+
print(speaker_ids)
|
| 206 |
+
print("speakers:")
|
| 207 |
+
print(speakers)
|
| 208 |
+
|
| 209 |
if t == "vits" or t == "vits-fast":
|
| 210 |
models_tts.append((name, author, cover_path, speakers, lang, example,
|
| 211 |
hps.symbols, create_tts_fn(model, hps, speaker_ids),
|