|
|
@ -20,6 +20,8 @@ ENC_MODELS_DIRT = f"encoder{os.sep}saved_models"
|
|
|
|
VOC_MODELS_DIRT = f"vocoder{os.sep}saved_models"
|
|
|
|
VOC_MODELS_DIRT = f"vocoder{os.sep}saved_models"
|
|
|
|
TEMP_SOURCE_AUDIO = f"wavs{os.sep}temp_source.wav"
|
|
|
|
TEMP_SOURCE_AUDIO = f"wavs{os.sep}temp_source.wav"
|
|
|
|
TEMP_RESULT_AUDIO = f"wavs{os.sep}temp_result.wav"
|
|
|
|
TEMP_RESULT_AUDIO = f"wavs{os.sep}temp_result.wav"
|
|
|
|
|
|
|
|
if not os.path.isdir("wavs"):
|
|
|
|
|
|
|
|
os.makedirs("wavs")
|
|
|
|
|
|
|
|
|
|
|
|
# Load local sample audio as options TODO: load dataset
|
|
|
|
# Load local sample audio as options TODO: load dataset
|
|
|
|
if os.path.isdir(AUDIO_SAMPLES_DIR):
|
|
|
|
if os.path.isdir(AUDIO_SAMPLES_DIR):
|
|
|
@ -44,6 +46,7 @@ else:
|
|
|
|
raise Exception(f"Model folder {VOC_MODELS_DIRT} doesn't exist.")
|
|
|
|
raise Exception(f"Model folder {VOC_MODELS_DIRT} doesn't exist.")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class Input(BaseModel):
|
|
|
|
class Input(BaseModel):
|
|
|
|
message: str = Field(
|
|
|
|
message: str = Field(
|
|
|
|
..., example="欢迎使用工具箱, 现已支持中文输入!", alias="文本内容"
|
|
|
|
..., example="欢迎使用工具箱, 现已支持中文输入!", alias="文本内容"
|
|
|
|