tobiccino commited on
Commit
5a4e089
1 Parent(s): 557399e

add stop duration option

Browse files
Files changed (1) hide show
  1. app.py +12 -11
app.py CHANGED
@@ -6,16 +6,17 @@ import gradio as gr
6
  import os
7
 
8
 
9
- def text_to_speech(text):
10
  print("starting")
11
  # prevent too long text
12
  if len(text) > 500:
13
  text = text[:500]
 
14
  text = nat_normalize_text(text)
15
  mel = text2mel(
16
  text,
17
  "lexicon.txt",
18
- 0.2,
19
  "acoustic_latest_ckpt.pickle",
20
  "duration_latest_ckpt.pickle",
21
  )
@@ -27,25 +28,25 @@ def text_to_speech(text):
27
  return (wave * (2**15)).astype(np.int16)
28
 
29
 
30
- def speak(text):
31
- y = text_to_speech(text)
32
  return 16_000, y
33
 
34
 
35
  title = "SLT TTS"
36
- description = "A vietnamese text-to-speech demo."
37
 
38
  gr.Interface(
39
  fn=speak,
40
- inputs="text",
41
  outputs="audio",
42
  title = title,
43
  examples = [
44
- "Trăm năm trong cõi người ta, chữ tài chữ mệnh khéo ghét nhau.",
45
- "Đoạn trường tân thanh, thường được biết đến với cái tên đơn giản Truyện Kiều, một truyện thơ của đại thi hào Nguyễn Du",
46
- "Lục Vân Tiên quê huyện Đông Thành, khôi ngô tuấn tú, tài kiêm văn võ. Nghe tin triều đình mở khoa thi, Vân Tiên từ giã thầy xuống núi đua tài.",
47
- " Quý Đôn, tên thuở nhỏ Danh Phương, vị quan thời trung hưng, cũng nhà thơ được mệnh danh nhà bác học lớn của Việt Nam trong thời phong kiến",
48
- "Tất cả mọi người đều sinh ra quyền bình đẳng. Tạo hóa cho họ những quyền không ai thể xâm phạm được; trong những quyền ấy, quyền được sống, quyền tự do quyền mưu cầu hạnh phúc."
49
  ],
50
  description=description,
51
  theme="default",
 
6
  import os
7
 
8
 
9
+ def text_to_speech(text,stop_duration_text):
10
  print("starting")
11
  # prevent too long text
12
  if len(text) > 500:
13
  text = text[:500]
14
+ float stop_duration_float=Float.parseFloat(stop_duration_text);
15
  text = nat_normalize_text(text)
16
  mel = text2mel(
17
  text,
18
  "lexicon.txt",
19
+ stop_duration_float,
20
  "acoustic_latest_ckpt.pickle",
21
  "duration_latest_ckpt.pickle",
22
  )
 
28
  return (wave * (2**15)).astype(np.int16)
29
 
30
 
31
+ def speak(text,stop_duration_text):
32
+ y = text_to_speech(text,stop_duration_text)
33
  return 16_000, y
34
 
35
 
36
  title = "SLT TTS"
37
+ description = "SLT Vietnamese Text to speech demo."
38
 
39
  gr.Interface(
40
  fn=speak,
41
+ inputs=["text","stop_duration_text"],
42
  outputs="audio",
43
  title = title,
44
  examples = [
45
+ "Thành phố muốn thí điểm thu thuế bất động sản thứ 2, tự quyết nhiều quyết định đầu để thu hút nguồn vốn tư nhân",
46
+ "Thứ năm, 22/12/2022, 10:59 những mét hàng rào đầu tiên quanh công viên lớn nhất thủ đô được tháo dỡ, chuyển công viên sang hình thức mở, không thu vé.",
47
+ " chú bác nhà em không học hành vẫn làm bất động sản. phải làm ngành này thì chỉ cần biết sơ sơ, không cần học hành?... Đó là vấn đề được học sinh trường Trung học phổ thông Gia Định, Thành phố Hồ Chí Minh đặt ra tại buổi khai mạc chương trình tư vấn tuyển sinh do Tạp chí Giáo dục Thành phố Hồ Chí Minh tổ chức sáng mồng 9 tháng 1.",
48
+ "Trong chuyến công du nước ngoài thứ hai từ khi Nga phát động chiến dịch quân sự tại Ukraine gần 12 tháng trước, Tổng thống Volodymyr Zelensky dự kiến phát biểu tại Nghị viện châu Âu ngày 9/2",
49
+ "Tuy nhiên, kế hoạch này đang đứng trước nguy đổ vỡ sau khi các quan chức Nghị viện châu âu làm rỉ kế hoạch mật tiếp đón Tổng thống Ukraine. Hồi đầu tuần, thông tin về chuyến thăm xuất hiện trên nhiều tờ báo châu âu, khởi nguồn từ báo La Stampa của Italy."
50
  ],
51
  description=description,
52
  theme="default",