Fix: add missing synthesize() method to BlueTTS class
Browse files
app.py
CHANGED
|
@@ -310,6 +310,10 @@ class BlueTTS:
|
|
| 310 |
if i < len(chunks) - 1: parts.append(silence)
|
| 311 |
return (np.concatenate(parts) if parts else np.array([], dtype=np.float32)), self.sample_rate
|
| 312 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 313 |
def _run(self, sess: ort.InferenceSession, feed: dict, model_name: str):
|
| 314 |
keys = self._model_keys.get(model_name)
|
| 315 |
if keys: feed = {**feed, **keys}
|
|
|
|
| 310 |
if i < len(chunks) - 1: parts.append(silence)
|
| 311 |
return (np.concatenate(parts) if parts else np.array([], dtype=np.float32)), self.sample_rate
|
| 312 |
|
| 313 |
+
def synthesize(self, text: str, lang: str = "he") -> Tuple[np.ndarray, int]:
|
| 314 |
+
phonemes = self._text_proc.phonemize(text, lang=lang)
|
| 315 |
+
return self.create(phonemes, lang=lang)
|
| 316 |
+
|
| 317 |
def _run(self, sess: ort.InferenceSession, feed: dict, model_name: str):
|
| 318 |
keys = self._model_keys.get(model_name)
|
| 319 |
if keys: feed = {**feed, **keys}
|