lj1995 commited on
Commit
400bc4b
·
verified ·
1 Parent(s): 6d40265

Update inference_webui.py

Browse files
Files changed (1) hide show
  1. inference_webui.py +6 -6
inference_webui.py CHANGED
@@ -211,7 +211,7 @@ def change_sovits_weights(sovits_path, prompt_language=None, text_language=None)
211
  print(vq_model.load_state_dict(dict_s2["weight"], strict=False))
212
  dict_language = dict_language_v1 if version == "v1" else dict_language_v2
213
  if prompt_language is not None and text_language is not None:
214
- if prompt_language in list(dict_languages()):
215
  prompt_text_update, prompt_language_update = (
216
  {"__type__": "update"},
217
  {"__type__": "update", "value": prompt_language},
@@ -219,14 +219,14 @@ def change_sovits_weights(sovits_path, prompt_language=None, text_language=None)
219
  else:
220
  prompt_text_update = {"__type__": "update", "value": ""}
221
  prompt_language_update = {"__type__": "update", "value": i18n("中文")}
222
- if text_language in list(dict_languages()):
223
  text_update, text_language_update = {"__type__": "update"}, {"__type__": "update", "value": text_language}
224
  else:
225
  text_update = {"__type__": "update", "value": ""}
226
  text_language_update = {"__type__": "update", "value": i18n("中文")}
227
  return (
228
- {"__type__": "update", "choices": list(dict_languages())},
229
- {"__type__": "update", "choices": list(dict_languages())},
230
  prompt_text_update,
231
  prompt_language_update,
232
  text_update,
@@ -798,7 +798,7 @@ with gr.Blocks(
798
  ),
799
  )
800
  prompt_language = gr.Dropdown(
801
- label=i18n("参考音频的语种"), choices=list(dict_languages()), value=i18n("中文")
802
  )
803
  inp_refs = gr.File(
804
  label=i18n(
@@ -813,7 +813,7 @@ with gr.Blocks(
813
  with gr.Column():
814
  text_language = gr.Dropdown(
815
  label=i18n("需要合成的语种。限制范围越小判别效果越好。"),
816
- choices=list(dict_languages()),
817
  value=i18n("中文"),
818
  )
819
  how_to_cut = gr.Dropdown(
 
211
  print(vq_model.load_state_dict(dict_s2["weight"], strict=False))
212
  dict_language = dict_language_v1 if version == "v1" else dict_language_v2
213
  if prompt_language is not None and text_language is not None:
214
+ if prompt_language in list(dict_language.keys()()):
215
  prompt_text_update, prompt_language_update = (
216
  {"__type__": "update"},
217
  {"__type__": "update", "value": prompt_language},
 
219
  else:
220
  prompt_text_update = {"__type__": "update", "value": ""}
221
  prompt_language_update = {"__type__": "update", "value": i18n("中文")}
222
+ if text_language in list(dict_language.keys()()):
223
  text_update, text_language_update = {"__type__": "update"}, {"__type__": "update", "value": text_language}
224
  else:
225
  text_update = {"__type__": "update", "value": ""}
226
  text_language_update = {"__type__": "update", "value": i18n("中文")}
227
  return (
228
+ {"__type__": "update", "choices": list(dict_language.keys()())},
229
+ {"__type__": "update", "choices": list(dict_language.keys()())},
230
  prompt_text_update,
231
  prompt_language_update,
232
  text_update,
 
798
  ),
799
  )
800
  prompt_language = gr.Dropdown(
801
+ label=i18n("参考音频的语种"), choices=list(dict_language.keys()()), value=i18n("中文")
802
  )
803
  inp_refs = gr.File(
804
  label=i18n(
 
813
  with gr.Column():
814
  text_language = gr.Dropdown(
815
  label=i18n("需要合成的语种。限制范围越小判别效果越好。"),
816
+ choices=list(dict_language.keys()()),
817
  value=i18n("中文"),
818
  )
819
  how_to_cut = gr.Dropdown(