feat: show outputs location in verbose mode
This commit is contained in:
parent
8cf2827b24
commit
06c1217e22
1 changed files with 2 additions and 6 deletions
8
tts.py
8
tts.py
|
@ -7,13 +7,10 @@ import torch
|
||||||
import argparse
|
import argparse
|
||||||
from kokoro import KPipeline
|
from kokoro import KPipeline
|
||||||
import soundfile as sf
|
import soundfile as sf
|
||||||
# import vlc
|
|
||||||
from tqdm import tqdm
|
from tqdm import tqdm
|
||||||
import pyperclip
|
import pyperclip
|
||||||
from yaspin import yaspin
|
from yaspin import yaspin
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
# See voices: https://huggingface.co/hexgrad/Kokoro-82M/blob/main/VOICES.md
|
# See voices: https://huggingface.co/hexgrad/Kokoro-82M/blob/main/VOICES.md
|
||||||
voices = {
|
voices = {
|
||||||
'pro': 'af_heart',
|
'pro': 'af_heart',
|
||||||
|
@ -168,12 +165,11 @@ def main():
|
||||||
start_time = time()
|
start_time = time()
|
||||||
output_files = generate_audio(generator, name, voice)
|
output_files = generate_audio(generator, name, voice)
|
||||||
generation_time = time() - start_time
|
generation_time = time() - start_time
|
||||||
|
directory,f = os.path.split(output_files[0])
|
||||||
|
|
||||||
if args.verbose:
|
if args.verbose:
|
||||||
print(f"[TTS] {len(output_files)} chunks generated in {generation_time:.2f} seconds")
|
print(f"[TTS] {len(output_files)} chunks generated in {generation_time:.2f} seconds")
|
||||||
print("[TTS] Now playing generated audio...")
|
print(f"[TTS] Output files are in: {directory}/*")
|
||||||
|
|
||||||
directory,f = os.path.split(output_files[0])
|
|
||||||
|
|
||||||
if args.skip_play:
|
if args.skip_play:
|
||||||
print(f"Audio player disabled: {directory}/*")
|
print(f"Audio player disabled: {directory}/*")
|
||||||
|
|
Loading…
Reference in a new issue