You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
76 lines
2.4 KiB
76 lines
2.4 KiB
3 years ago
|
# Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved.
|
||
|
|
#
|
||
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
# you may not use this file except in compliance with the License.
|
||
|
|
# You may obtain a copy of the License at
|
||
|
|
#
|
||
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
#
|
||
|
|
# Unless required by applicable law or agreed to in writing, software
|
||
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
# See the License for the specific language governing permissions and
|
||
|
|
# limitations under the License.
|
||
|
|
import argparse
|
||
|
|
import json
|
||
|
|
import time
|
||
|
|
|
||
|
|
import requests
|
||
|
|
|
||
|
|
from paddlespeech.cli.log import logger
|
||
|
|
|
||
|
|
|
||
|
|
# Request and response
|
||
|
|
def text_client(args):
|
||
|
|
""" Request and response
|
||
|
|
Args:
|
||
|
|
text: A sentence to be processed by PaddleSpeech Text Server
|
||
|
|
outfile: The punctuation text
|
||
|
|
"""
|
||
|
|
url = "http://" + str(args.server) + ":" + str(
|
||
|
|
args.port) + "/paddlespeech/text"
|
||
|
|
request = {
|
||
|
|
"text": args.text,
|
||
|
|
}
|
||
|
|
|
||
|
|
response = requests.post(url, json.dumps(request))
|
||
|
|
response_dict = response.json()
|
||
|
|
punc_text = response_dict["result"]["punc_text"]
|
||
|
|
|
||
|
|
# transform audio
|
||
|
|
outfile = args.output
|
||
|
|
if outfile:
|
||
|
|
with open(outfile, 'w') as w:
|
||
|
|
w.write(punc_text + "\n")
|
||
|
|
|
||
|
|
logger.info(f"The punc text is: {punc_text}")
|
||
|
|
return punc_text
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
parser.add_argument(
|
||
|
|
'--text',
|
||
|
|
type=str,
|
||
|
|
default="今天的天气真不错啊你下午有空吗我想约你一起去吃饭",
|
||
|
|
help='A sentence to be synthesized')
|
||
|
|
parser.add_argument(
|
||
|
|
'--output', type=str, default="./punc_text", help='Punc text file')
|
||
|
|
parser.add_argument(
|
||
|
|
"--server", type=str, help="server ip", default="127.0.0.1")
|
||
|
|
parser.add_argument("--port", type=int, help="server port", default=8090)
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
st = time.time()
|
||
|
|
try:
|
||
|
|
punc_text = text_client(args)
|
||
|
|
time_consume = time.time() - st
|
||
|
|
time_per_word = time_consume / len(args.text)
|
||
|
|
print("Text Process successfully.")
|
||
|
|
print("Inference time: %f" % (time_consume))
|
||
|
|
print("The text length: %f" % (len(args.text)))
|
||
|
|
print("The time per work is: %f" % (time_per_word))
|
||
|
|
except BaseException as e:
|
||
|
|
logger.info("Failed to Process text.")
|
||
|
|
logger.info(e)
|