matias commited on
Commit
98c6a7f
1 Parent(s): 6e5a5d5

added makewaveform

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -18,9 +18,14 @@ def generate_sound(text):
18
  print(text)
19
  text=translate_text(text)
20
  print(text)
21
- audio = pipe(text, num_inference_steps=10, audio_length_in_s=5.0).audios[0]
 
 
 
22
  rate =160000
23
- return rate, audio
 
 
24
  def translate_text(text):
25
  translator = Translator()
26
  translated_text=translator.translate(text, src='es',dest="en")
 
18
  print(text)
19
  text=translate_text(text)
20
  print(text)
21
+ waveforms = pipe(text,
22
+ num_inference_steps=25,
23
+ audio_length_in_s=5,
24
+ negative_prompt = "low quality, average quality").audios
25
  rate =160000
26
+ #return rate, audio
27
+ return gr.make_waveform((rate, waveforms[0]))
28
+
29
  def translate_text(text):
30
  translator = Translator()
31
  translated_text=translator.translate(text, src='es',dest="en")