Gregniuki commited on
Commit
311bb00
·
1 Parent(s): 5dc78e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -2
app.py CHANGED
@@ -129,7 +129,7 @@ def detect_onnx_models(path):
129
  def get_initial_values():
130
  # You can set default values or load them from a configuration file here
131
  selected_model = onnx_models[0] if onnx_models else "default_model"
132
- selected_speaker_id = "default_speaker_id" # Default value
133
 
134
  # Check if there are onnx models and load the speaker_id_map from the first model's config
135
  if onnx_models:
@@ -138,6 +138,8 @@ def get_initial_values():
138
  speaker_id_map = first_model_config.get("speaker_id_map")
139
  if speaker_id_map:
140
  selected_speaker_id = next(iter(speaker_id_map)) # Get the first speaker_id
 
 
141
 
142
  return selected_model, selected_speaker_id
143
  @app.get("/get_speaker_id_map")
@@ -147,7 +149,11 @@ async def get_speaker_id_map(selected_model: str):
147
  print("selectef config ",config)
148
  if config:
149
  speaker_id_map = config.get("speaker_id_map", {})
150
- return {"speaker_id_map": speaker_id_map}
 
 
 
 
151
 
152
  # Handle the case where the config is not available for the selected model
153
  return {"speaker_id_map": {}}
@@ -235,6 +241,7 @@ async def main(
235
  ]
236
  sess_options = onnxruntime.SessionOptions()
237
  model, config = load_onnx(selected_model, sess_options, providers)
 
238
  speaker_id_map = config.get("speaker_id_map", {})
239
 
240
  auto_play = play
 
129
  def get_initial_values():
130
  # You can set default values or load them from a configuration file here
131
  selected_model = onnx_models[0] if onnx_models else "default_model"
132
+ selected_speaker_id = 0 # Default value
133
 
134
  # Check if there are onnx models and load the speaker_id_map from the first model's config
135
  if onnx_models:
 
138
  speaker_id_map = first_model_config.get("speaker_id_map")
139
  if speaker_id_map:
140
  selected_speaker_id = next(iter(speaker_id_map)) # Get the first speaker_id
141
+ else:
142
+ selected_speaker_id = 0
143
 
144
  return selected_model, selected_speaker_id
145
  @app.get("/get_speaker_id_map")
 
149
  print("selectef config ",config)
150
  if config:
151
  speaker_id_map = config.get("speaker_id_map", {})
152
+ if speaker_id_map:
153
+ return {"speaker_id_map": speaker_id_map}
154
+ else:
155
+ speaker_id_map = {"speaker1": "0",
156
+ }
157
 
158
  # Handle the case where the config is not available for the selected model
159
  return {"speaker_id_map": {}}
 
241
  ]
242
  sess_options = onnxruntime.SessionOptions()
243
  model, config = load_onnx(selected_model, sess_options, providers)
244
+ config["espeak"]["voice"] = speaker
245
  speaker_id_map = config.get("speaker_id_map", {})
246
 
247
  auto_play = play