thewh1teagle commited on
Commit
11d41f5
1 Parent(s): 3ae95eb

update app

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -10,6 +10,7 @@ python3 app.py
10
 
11
  import gradio as gr
12
  from israwave import IsraWave
 
13
  from nakdimon_ort import Nakdimon
14
  from israwave.segment import SegmentExtractor
15
  import numpy as np
@@ -19,7 +20,8 @@ speech_model = IsraWave('israwave.onnx', 'espeak-ng-data')
19
  niqqud_model = Nakdimon('nakdimon.onnx')
20
 
21
  def create_audio(text: str, rate, pitch, energy):
22
- text = niqqud_model.compute(text)
 
23
  waveforms = []
24
  for segment in segment_extractor.extract_segments(text):
25
  waveform = speech_model.create(segment.text, rate=rate, pitch=pitch, energy=energy)
 
10
 
11
  import gradio as gr
12
  from israwave import IsraWave
13
+ from israwave.helpers import text_has_niqqud
14
  from nakdimon_ort import Nakdimon
15
  from israwave.segment import SegmentExtractor
16
  import numpy as np
 
20
  niqqud_model = Nakdimon('nakdimon.onnx')
21
 
22
  def create_audio(text: str, rate, pitch, energy):
23
+ if not text_has_niqqud(text):
24
+ text = niqqud_model.compute(text)
25
  waveforms = []
26
  for segment in segment_extractor.extract_segments(text):
27
  waveform = speech_model.create(segment.text, rate=rate, pitch=pitch, energy=energy)