sync ui

app.py

@@ -10,16 +10,17 @@ python3 app.py
 
 import gradio as gr
 from israwave import IsraWave
-from israwave.helpers import text_has_niqqud
+from israwave.helpers import text_has_niqqud, float_to_int16
 from nakdimon_ort import Nakdimon
 from israwave.segment import SegmentExtractor
 import numpy as np
 
 segment_extractor = SegmentExtractor()
-speech_model = IsraWave('israwave.onnx', 'espeak-ng-data')
-niqqud_model = Nakdimon('nakdimon.onnx')
+speech_model = IsraWave("israwave.onnx", "espeak-ng-data")
+niqqud_model = Nakdimon("nakdimon.onnx")
 
-def create_audio(text: str, rate, pitch, energy):
+
+def create(text: str, rate, pitch, energy):
     if not text_has_niqqud(text):
         text = niqqud_model.compute(text)
     waveforms = []
@@ -28,11 +29,10 @@ def create_audio(text: str, rate, pitch, energy):
         waveforms.append(waveform.samples)
         silence = segment.create_pause(waveform.sample_rate)
         waveforms.append(silence)
-    return np.concatenate(waveforms), speech_model.sample_rate
-
-def create(text, rate, pitch, energy):
-    samples, sample_rate = create_audio(text, rate, pitch, energy)
-    return (sample_rate, samples)
+    waveform = np.concatenate(waveforms)
+    # Gradio expect int16
+    waveform = float_to_int16(waveform)
+    return speech_model.sample_rate, waveform
 
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
@@ -41,16 +41,21 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     <h1 style='text-align: center;'>IsraWave</h1>
     <p style='text-align: center;'>Text-to-Speech model for Hebrew</p>
     """)
-    
+
     # Use Textarea with RTL direction
-    text = gr.TextArea(label="text", lines=4, elem_id="rtl_textarea", value='זה כיף להזמין דברים באינטרנט, אבל הרבה פחות כיף לחכות ולחכות עד שהם יגיעו אלינו. אז מה בעצם עובר על החבילות בדרך הארוכה עד לבית שלנו? והאם אפשר לגרום לכך שהן יגיעו מהר יותר? ')
+    text = gr.TextArea(
+        label="text",
+        lines=4,
+        elem_id="rtl_textarea",
+        value="זה כיף להזמין דברים באינטרנט, אבל הרבה פחות כיף לחכות ולחכות עד שהם יגיעו אלינו. אז מה בעצם עובר על החבילות בדרך הארוכה עד לבית שלנו? והאם אפשר לגרום לכך שהן יגיעו מהר יותר? ",
+    )
     rate = gr.Slider(0.1, 10, label="rate", value=1.0)
     pitch = gr.Slider(0.1, 10, label="pitch", value=1.0)
     energy = gr.Slider(0.1, 10, label="energy", value=1.0)
 
-    button = gr.Button("Create", elem_id="create_button")
+    button = gr.Button("Create")
     output = gr.Audio()
-    
+
     button.click(fn=create, inputs=[text, rate, pitch, energy], outputs=output)
 
     # Custom CSS for RTL direction
@@ -60,9 +65,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         font-size: 20px;
     }
     """
-    
+
     gr.Markdown("""
     <p style='text-align: center;'><a href='https://github.com/thewh1teagle/israwave' target='_blank'>Israwave on Github</a></p>
     """)
 
-demo.launch()
+demo.launch()

requirements.txt

@@ -1,2 +1,2 @@
-gradio
-israwave
+gradio==4.44.0
+israwave==0.1.5