chore: update requirements

This commit is contained in:
Ayo Ayco 2025-09-03 21:21:49 +02:00
parent e5c9aae613
commit 973dcd5d78
2 changed files with 5 additions and 5 deletions

View file

@ -1,11 +1,10 @@
from kokoro import KPipeline from kokoro import KPipeline
from IPython.display import display, Audio
import soundfile as sf import soundfile as sf
import torch
import sys import sys
import os import os
import vlc import vlc
from time import sleep from time import sleep
from tqdm import tqdm
pipeline = KPipeline(lang_code='a', device='xpu') pipeline = KPipeline(lang_code='a', device='xpu')
@ -23,8 +22,8 @@ output_files = []
for i, (gs, ps, audio) in enumerate(generator): for i, (gs, ps, audio) in enumerate(generator):
# print(i, gs, ps) # print(i, gs, ps)
display(Audio(data=audio, rate=24000, autoplay=i==0)) output_file_name=f'outputs/{name}-{i}.wav'
output_file_name=f'{name}-{i}.wav' os.makedirs(os.path.dirname(output_file_name), exist_ok=True)
print(f"Done generating audio: {output_file_name}") print(f"Done generating audio: {output_file_name}")
sf.write(output_file_name, audio, 24000) sf.write(output_file_name, audio, 24000)
output_files.append(output_file_name) output_files.append(output_file_name)
@ -37,4 +36,5 @@ for output in output_files:
sleep(0.1) sleep(0.1)
duration=media.get_length() / 1000 duration=media.get_length() / 1000
print(f"duration: {duration}s") print(f"duration: {duration}s")
#for i in tqdm(range(100)):
sleep(duration) sleep(duration)

View file

@ -1,5 +1,5 @@
kokoro kokoro
IPython
soundfile soundfile
torch torch
python-vlc python-vlc
tqdm