chnk58hoang commited on
Commit
5ab552b
1 Parent(s): 2a8bf06

cleancode follow PEP8

Browse files
Files changed (2) hide show
  1. app.py +4 -5
  2. utils.py +12 -3
app.py CHANGED
@@ -6,12 +6,14 @@ from utils import load_model, normalize_text
6
  vits_model = load_model()
7
  vits_model.tts('Alo')
8
 
 
9
  def text_to_speech(text):
10
  text = normalize_text(text)
11
  audio = vits_model.tts(text)
12
 
13
  audio = np.array(audio)
14
- return 16000,audio
 
15
 
16
  gr.Interface(
17
  fn=text_to_speech,
@@ -21,11 +23,8 @@ gr.Interface(
21
  examples=[
22
  "Trăm năm trong cõi người ta, chữ tài chữ mệnh khéo là ghét nhau.",
23
  "Đoạn trường tân thanh, thường được biết đến với cái tên đơn giản là Truyện Kiều, là một truyện thơ của đại thi hào Nguyễn Du",
24
- "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.",
25
- "Lê Quý Đôn, tên thuở nhỏ là Lê Danh Phương, là vị quan thời Lê trung hưng, cũng là nhà thơ và được mệnh danh là nhà bác học lớn của Việt Nam trong thời phong kiến",
26
- "Tất cả mọi người đều sinh ra có quyền bình đẳng. Tạo hóa cho họ những quyền không ai có thể xâm phạm được; trong những quyền ấy, có quyền được sống, quyền tự do và quyền mưu cầu hạnh phúc.",
27
  ],
28
  theme="default",
29
  ).launch(debug=False)
30
 
31
-
 
6
  vits_model = load_model()
7
  vits_model.tts('Alo')
8
 
9
+
10
  def text_to_speech(text):
11
  text = normalize_text(text)
12
  audio = vits_model.tts(text)
13
 
14
  audio = np.array(audio)
15
+ return 16000, audio
16
+
17
 
18
  gr.Interface(
19
  fn=text_to_speech,
 
23
  examples=[
24
  "Trăm năm trong cõi người ta, chữ tài chữ mệnh khéo là ghét nhau.",
25
  "Đoạn trường tân thanh, thường được biết đến với cái tên đơn giản là Truyện Kiều, là một truyện thơ của đại thi hào Nguyễn Du",
26
+ "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."
 
 
27
  ],
28
  theme="default",
29
  ).launch(debug=False)
30
 
 
utils.py CHANGED
@@ -3,12 +3,16 @@ import unicodedata
3
  import regex
4
 
5
 
6
-
7
-
8
  num_re = regex.compile(r"([0-9.,]*[0-9])")
9
  digits = ["không", "một", "hai", "ba", "bốn", "năm", "sáu", "bảy", "tám", "chín"]
10
 
 
11
  def read_number(num: str) -> str:
 
 
 
 
 
12
  if len(num) == 1:
13
  return digits[int(num)]
14
  elif len(num) == 2 and num.isdigit():
@@ -74,6 +78,11 @@ def load_model():
74
 
75
 
76
  def normalize_text(text):
 
 
 
 
 
77
  # lowercase
78
  text = text.lower()
79
  # unicode normalize
@@ -85,7 +94,7 @@ def normalize_text(text):
85
  text = text.replace("!", "")
86
  text = text.replace("?", "")
87
  text = text.replace("(", "")
88
-
89
  text = num_re.sub(r" \1 ", text)
90
  words = text.split()
91
  words = [read_number(w) if num_re.fullmatch(w) else w for w in words]
 
3
  import regex
4
 
5
 
 
 
6
  num_re = regex.compile(r"([0-9.,]*[0-9])")
7
  digits = ["không", "một", "hai", "ba", "bốn", "năm", "sáu", "bảy", "tám", "chín"]
8
 
9
+
10
  def read_number(num: str) -> str:
11
+ """Translate numeric text into written form
12
+
13
+ Args: num (str) numeric text
14
+ Returns: (str) written form of num
15
+ """
16
  if len(num) == 1:
17
  return digits[int(num)]
18
  elif len(num) == 2 and num.isdigit():
 
78
 
79
 
80
  def normalize_text(text):
81
+ """Normalize the input text
82
+
83
+ Args: text (str) the input text
84
+ Returns: text (str) the normalized text
85
+ """
86
  # lowercase
87
  text = text.lower()
88
  # unicode normalize
 
94
  text = text.replace("!", "")
95
  text = text.replace("?", "")
96
  text = text.replace("(", "")
97
+ # Convert numeric text into written form
98
  text = num_re.sub(r" \1 ", text)
99
  words = text.split()
100
  words = [read_number(w) if num_re.fullmatch(w) else w for w in words]