mrfakename
commited on
Commit
•
012ec7f
1
Parent(s):
e315c87
Sync from GitHub repo
Browse filesThis Space is synced from the GitHub repo: https://github.com/SWivid/F5-TTS. Please submit contributions to the Space there
- inference-cli.py +3 -7
inference-cli.py
CHANGED
@@ -150,14 +150,10 @@ def main_process(ref_audio, ref_text, text_gen, model_obj, remove_silence):
|
|
150 |
reg2 = r'\[(\w+)\]'
|
151 |
for text in chunks:
|
152 |
match = re.match(reg2, text)
|
153 |
-
if match:
|
154 |
-
voice = match[1]
|
155 |
-
else:
|
156 |
-
print("No voice tag found, using main.")
|
157 |
-
voice = "main"
|
158 |
-
if voice not in voices:
|
159 |
-
print(f"Voice {voice} not found, using main.")
|
160 |
voice = "main"
|
|
|
|
|
161 |
text = re.sub(reg2, "", text)
|
162 |
gen_text = text.strip()
|
163 |
ref_audio = voices[voice]['ref_audio']
|
|
|
150 |
reg2 = r'\[(\w+)\]'
|
151 |
for text in chunks:
|
152 |
match = re.match(reg2, text)
|
153 |
+
if not match or voice not in voices:
|
|
|
|
|
|
|
|
|
|
|
|
|
154 |
voice = "main"
|
155 |
+
else:
|
156 |
+
voice = match[1]
|
157 |
text = re.sub(reg2, "", text)
|
158 |
gen_text = text.strip()
|
159 |
ref_audio = voices[voice]['ref_audio']
|