-
Notifications
You must be signed in to change notification settings - Fork 200
/
test.py
81 lines (64 loc) · 2.14 KB
/
test.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
#!usr/bin/env python
# -*- coding: utf-8 -*-
# author: kuangdd
# date: 2020/2/23
"""
test
"""
from pathlib import Path
import logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(Path(__file__).stem)
def test_sdk_api():
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '_'
from ttskit import sdk_api
text = '初始化。'
wav = sdk_api.tts_sdk(text, speaker='biaobei', audio='24', processes=0)
text = '欢迎使用语音合成工具箱这是二十个字的文本。' * 5
wav = sdk_api.tts_sdk(text, speaker='biaobei', audio='24', processes=0)
def test_cli_api():
from ttskit import cli_api
args = cli_api.parse_args()
cli_api.tts_cli(args)
def test_web_api():
from ttskit import web_api
web_api.app.run(host='0.0.0.0', port=2718, debug=False, processes=6, threaded=False)
# 用POST或GET方法请求:http://localhost:2718/tts,传入参数text、audio、speaker。
# 例如GET方法请求:http://localhost:2718/tts?text=这是个例子&audio=2
def test_http_server():
from ttskit import http_server
from multiprocessing import Process
server = http_server.start_sever()
# 单进程
# server.serve_forever()
# 多进程
server.start()
for i in range(6):
p = Process(target=http_server.serve_forever, args=(server,))
p.start()
# 打开网页:http://localhost:9000/ttskit
if __name__ == "__main__":
logger.info(__file__)
from ttskit.sdk_api import load_models, _global_mspk_kwargs
load_models(**_global_mspk_kwargs)
# test_sdk_api()
# test_cli_api()
# test_web_api()
# test_http_server()
# import time
# import threadpool
# def sayhello(str):
# print("Hello ",str)
# time.sleep(2)
# return 'hello'
#
# name_list =['xiaozi','aa','bb','cc']
# start_time = time.time()
# pool = threadpool.ThreadPool(10)
# requests = threadpool.makeRequests(sayhello, name_list)
# out = [pool.putRequest(req) for req in requests]
# print([req.args for req in requests])
# print(out, requests)
# pool.wait()
# print('%d second'% (time.time()-start_time))