SillyTavern/SillyTavern-extras
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern-Extras.git synced 2026-05-01 03:41:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move scipy import closer

Browse Source
This commit is contained in:
SillyLossy
2023-03-06 13:50:00 +02:00
parent 65f65ec42b
commit f8cb0d2beb
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
server.py
View File

@@ -10,7 +10,6 @@ import unicodedata
import torch import torch
import time import time
import numpy as np import numpy as np
from scipy.io import wavfile
from glob import glob from glob import glob
import json import json
import os import os
@@ -263,6 +262,7 @@ def generate_image(input: str, steps: int = 30, scale: int = 6) -> Image:
def generate_audio(text: str, voice: str): def generate_audio(text: str, voice: str):
from scipy.io import wavfile
audio = tts_model.tts(text=text, speaker_wav=f'tts_voices/{voice}.wav', language='en') audio = tts_model.tts(text=text, speaker_wav=f'tts_voices/{voice}.wav', language='en')
filename = f'tts_output/{voice}_{time.time_ns()}.wav'; filename = f'tts_output/{voice}_{time.time_ns()}.wav';
wavfile.write(filename, sample_rate, np.array(audio)) wavfile.write(filename, sample_rate, np.array(audio))