TTS/server/server.py

32 lines
993 B
Python
Raw Normal View History

2018-06-05 12:15:48 +00:00
#!flask/bin/python
import argparse
from synthesizer import Synthesizer
from TTS.utils.generic_utils import load_config
from flask import (Flask, Response, request,
render_template, send_file)
parser = argparse.ArgumentParser()
parser.add_argument('-c', '--config_path', type=str,
help='path to config file for training')
args = parser.parse_args()
config = load_config(args.config_path)
app = Flask(__name__)
synthesizer = Synthesizer()
synthesizer.load_model(config.model_path, config.model_name,
config.model_config, config.use_cuda)
@app.route('/')
def index():
return render_template('index.html')
@app.route('/api/tts', methods=['GET'])
def tts():
text = request.args.get('text')
print(" > Model input: {}".format(text))
data = synthesizer.tts(text)
2018-06-05 14:15:57 +00:00
return send_file(data,
2018-06-05 12:15:48 +00:00
mimetype='audio/wav')
if __name__ == '__main__':
app.run(debug=True, host='0.0.0.0', port=config.port)