improve error response format & docs

This commit is contained in:
Daiki Arai
2023-12-31 00:15:40 +09:00
parent 55d4bbf783
commit ef6a429421
2 changed files with 69 additions and 92 deletions

10
app.py
View File

@@ -3,6 +3,7 @@ import datetime
import os
import sys
import warnings
import enum
import gradio as gr
import numpy as np
@@ -15,7 +16,14 @@ from config import config
from infer import get_net_g, infer
from tools.log import logger
languages = ["JP", "EN", "ZH"]
class Languages(str, enum.Enum):
JP = "JP"
EN = "EN"
ZH = "ZH"
languages = [l.value for l in Languages]
DEFAULT_SDP_RATIO: float = 0.2
DEFAULT_NOISE: float = 0.6