Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
#
|
| 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 |
-
|
|
|
|
| 187 |
vits=vits_config,
|
| 188 |
num_threads=1,
|
| 189 |
debug=False,
|
| 190 |
provider="cpu",
|
| 191 |
)
|
| 192 |
|
| 193 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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:
|