Ikaros521 commited on
Commit
e5a4e96
·
1 Parent(s): 05af073

修复app.py代码错误

Browse files
Files changed (1) hide show
  1. app.py +9 -0
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),