randusertry commited on
Commit
bf98578
·
verified ·
1 Parent(s): 99eec13

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -4
app.py CHANGED
@@ -173,7 +173,7 @@ breton_engine = None
173
  def get_breton_engine():
174
  global breton_engine
175
  if breton_engine is None:
176
- # Configuration for Sherpa-ONNX VITS
177
  vits_config = sherpa_onnx.OfflineTtsVitsModelConfig(
178
  model=os.path.join(MODEL_DIR, "breton-model.onnx"),
179
  tokens=os.path.join(MODEL_DIR, "breton-tokens.txt"),
@@ -183,16 +183,26 @@ def get_breton_engine():
183
  length_scale=1.0,
184
  )
185
 
186
- config = sherpa_onnx.OfflineTtsModelConfig(
 
187
  vits=vits_config,
188
  num_threads=1,
189
  debug=False,
190
  provider="cpu",
191
  )
192
 
193
- breton_engine = sherpa_onnx.OfflineTts(config)
 
 
 
 
 
 
 
 
 
194
  return breton_engine
195
-
196
  @app.post("/tts/breton")
197
  async def get_breton_tts(request: TTSRequest):
198
  try:
 
173
  def get_breton_engine():
174
  global breton_engine
175
  if breton_engine is None:
176
+ # 1. Specific VITS model settings
177
  vits_config = sherpa_onnx.OfflineTtsVitsModelConfig(
178
  model=os.path.join(MODEL_DIR, "breton-model.onnx"),
179
  tokens=os.path.join(MODEL_DIR, "breton-tokens.txt"),
 
183
  length_scale=1.0,
184
  )
185
 
186
+ # 2. Wrap VITS into the Model Config
187
+ model_config = sherpa_onnx.OfflineTtsModelConfig(
188
  vits=vits_config,
189
  num_threads=1,
190
  debug=False,
191
  provider="cpu",
192
  )
193
 
194
+ # 3. Wrap everything into the Top-Level OfflineTtsConfig (The missing step!)
195
+ full_config = sherpa_onnx.OfflineTtsConfig(
196
+ model=model_config,
197
+ # rule_fsts is required for some models, empty string is fine here
198
+ rule_fsts="",
199
+ max_num_sentences=1,
200
+ )
201
+
202
+ # Now pass the full_config to the constructor
203
+ breton_engine = sherpa_onnx.OfflineTts(full_config)
204
  return breton_engine
205
+
206
  @app.post("/tts/breton")
207
  async def get_breton_tts(request: TTSRequest):
208
  try: