notmax123 commited on
Commit
7ab55a7
·
1 Parent(s): 19e4408

Fix: add missing synthesize() method to BlueTTS class

Browse files
Files changed (1) hide show
  1. app.py +4 -0
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}