tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Feat: support auto split in webui (#158)

Browse Source 
* Feat: support auto split in webui

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Sora
2023-11-08 17:08:47 +08:00
committed by GitHub
parent af7b0ea0ff
commit ee423f29df
3 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
server_fastapi.py
View File

@@ -5,6 +5,7 @@ import logging
import gc import gc
import random import random
import gradio
import numpy as np import numpy as np
import utils import utils
from fastapi import FastAPI, Query, Request from fastapi import FastAPI, Query, Request
@@ -245,6 +246,7 @@ if __name__ == "__main__":
) )
audios.append(np.zeros((int)(44100 * 0.3))) audios.append(np.zeros((int)(44100 * 0.3)))
audio = np.concatenate(audios) audio = np.concatenate(audios)
audio = gradio.processing_utils.convert_to_16_bit_wav(audio)
wavContent = BytesIO() wavContent = BytesIO()
wavfile.write( wavfile.write(
wavContent, loaded_models.models[model_id].hps.data.sampling_rate, audio wavContent, loaded_models.models[model_id].hps.data.sampling_rate, audio

4
tools/log.py
View File

@@ -9,6 +9,8 @@ import sys
logger.remove() logger.remove()
# 自定义格式并添加到标准输出 # 自定义格式并添加到标准输出
log_format = "<g>{time:MM-DD HH:mm:ss}</g> [<lvl>{level}</lvl>] | {message}" log_format = (
"<g>{time:MM-DD HH:mm:ss}</g> [<lvl>{level}</lvl>] | {file}:{line} | {message}"
)
logger.add(sys.stdout, format=log_format) logger.add(sys.stdout, format=log_format)