From 973dcd5d78c0a44159340fda03415720f9e4c260 Mon Sep 17 00:00:00 2001 From: Ayo Date: Wed, 3 Sep 2025 21:21:49 +0200 Subject: [PATCH] chore: update requirements --- main.py | 8 ++++---- requirements.txt | 2 +- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/main.py b/main.py index 01ec3cf..27ca52d 100644 --- a/main.py +++ b/main.py @@ -1,11 +1,10 @@ from kokoro import KPipeline -from IPython.display import display, Audio import soundfile as sf -import torch import sys import os import vlc from time import sleep +from tqdm import tqdm pipeline = KPipeline(lang_code='a', device='xpu') @@ -23,8 +22,8 @@ output_files = [] for i, (gs, ps, audio) in enumerate(generator): # print(i, gs, ps) - display(Audio(data=audio, rate=24000, autoplay=i==0)) - output_file_name=f'{name}-{i}.wav' + output_file_name=f'outputs/{name}-{i}.wav' + os.makedirs(os.path.dirname(output_file_name), exist_ok=True) print(f"Done generating audio: {output_file_name}") sf.write(output_file_name, audio, 24000) output_files.append(output_file_name) @@ -37,4 +36,5 @@ for output in output_files: sleep(0.1) duration=media.get_length() / 1000 print(f"duration: {duration}s") + #for i in tqdm(range(100)): sleep(duration) diff --git a/requirements.txt b/requirements.txt index 2dbaf95..dc2faf8 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ kokoro -IPython soundfile torch python-vlc +tqdm