mirror of https://github.com/coqui-ai/TTS.git
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
|
#!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)
|
||
|
|
return send_file(data,
|
||
|
|
attachment_filename="testing.wav",
|
||
|
|
as_attachment=True,
|
||
|
|
mimetype='audio/wav')
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
app.run(debug=True, host='0.0.0.0', port=config.port)
|