SillyTavern/SillyTavern-extras
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern-Extras.git synced 2026-03-06 12:00:31 +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 time
import numpy as np
from scipy.io import wavfile
from glob import glob
import json
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):
from scipy.io import wavfile
audio = tts_model.tts(text=text, speaker_wav=f'tts_voices/{voice}.wav', language='en')
filename = f'tts_output/{voice}_{time.time_ns()}.wav';
wavfile.write(filename, sample_rate, np.array(audio))