Update app.py
Browse files
app.py
CHANGED
|
@@ -11,7 +11,7 @@ import ONNXVITS_infer
|
|
| 11 |
app = Flask(__name__)
|
| 12 |
logging.getLogger('numba').setLevel(logging.WARNING)
|
| 13 |
|
| 14 |
-
|
| 15 |
"title": "Trilingual",
|
| 16 |
"model_path": "./pretrained_models/G_trilingual.pth",
|
| 17 |
"config_path": "./configs/uma_trilingual.json",
|
|
@@ -27,10 +27,10 @@ JAPANESE = {
|
|
| 27 |
|
| 28 |
models_tts = []
|
| 29 |
models_info = [
|
| 30 |
-
|
| 31 |
JAPANESE
|
| 32 |
]
|
| 33 |
-
MODEL = { "japanese": JAPANESE, "
|
| 34 |
|
| 35 |
def load_models():
|
| 36 |
for info in models_info:
|
|
@@ -88,7 +88,7 @@ def get_model_data(model):
|
|
| 88 |
|
| 89 |
@app.route("/")
|
| 90 |
def index():
|
| 91 |
-
return jsonify({
|
| 92 |
|
| 93 |
@app.route("/<model>/speakers", methods=["GET"])
|
| 94 |
def speakers(model):
|
|
|
|
| 11 |
app = Flask(__name__)
|
| 12 |
logging.getLogger('numba').setLevel(logging.WARNING)
|
| 13 |
|
| 14 |
+
TRILINGUAL = {
|
| 15 |
"title": "Trilingual",
|
| 16 |
"model_path": "./pretrained_models/G_trilingual.pth",
|
| 17 |
"config_path": "./configs/uma_trilingual.json",
|
|
|
|
| 27 |
|
| 28 |
models_tts = []
|
| 29 |
models_info = [
|
| 30 |
+
TRILINGUAL,
|
| 31 |
JAPANESE
|
| 32 |
]
|
| 33 |
+
MODEL = { "japanese": JAPANESE, "trilingual": TRILANGUAL }
|
| 34 |
|
| 35 |
def load_models():
|
| 36 |
for info in models_info:
|
|
|
|
| 88 |
|
| 89 |
@app.route("/")
|
| 90 |
def index():
|
| 91 |
+
return jsonify({"status": "OK" })
|
| 92 |
|
| 93 |
@app.route("/<model>/speakers", methods=["GET"])
|
| 94 |
def speakers(model):
|