chore: update requirements
This commit is contained in:
parent
e5c9aae613
commit
973dcd5d78
2 changed files with 5 additions and 5 deletions
8
main.py
8
main.py
|
@ -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)
|
||||||
|
|
|
@ -1,5 +1,5 @@
|
||||||
kokoro
|
kokoro
|
||||||
IPython
|
|
||||||
soundfile
|
soundfile
|
||||||
torch
|
torch
|
||||||
python-vlc
|
python-vlc
|
||||||
|
tqdm
|
||||||
|
|
Loading…
Reference in a new issue