diff --git a/app/ktv_adapter.py b/app/ktv_adapter.py index c7d3424..aad2165 100644 --- a/app/ktv_adapter.py +++ b/app/ktv_adapter.py @@ -6,14 +6,14 @@ KTV_FILE_BASE: 文件服务地址,默认同 KTV_API_BASE 所在域 步骤→模型映射(KTV_STEP_MODELS 字典,可通过环境变量覆盖各项): - demucs_separating → ky-demucs-separate - asr_transcribing → ky-asr-transcribe - subtitle_rendering → ky-subtitle-render - face_detecting → ky-face-detect - realesrgan_upscaling → ky-realesrgan-upscale - rvc_converting → ky-rvc-convert - video_evaluating → ky-video-eval-evaluate - songrate_evaluating → ky-songrate-evaluate + demucs_separating → demucs-separate + asr_transcribing → asr-transcribe + subtitle_rendering → subtitle-render + face_detecting → face-detect + realesrgan_upscaling → realesrgan-upscale + rvc_converting → rvc-convert + video_evaluating → video-eval-evaluate + songrate_evaluating → songrate-evaluate 本地操作(ffmpeg/ffprobe)不走 API。 """ @@ -40,14 +40,14 @@ API_KEY = os.environ.get("KTV_API_KEY", "") FILE_BASE = os.environ.get("KTV_FILE_BASE", API_BASE.rsplit("/", 2)[0]) STEP_MODELS = { - "demucs_separating": os.environ.get("KTV_MODEL_DEMUCS", "ky-demucs-separate"), - "asr_transcribing": os.environ.get("KTV_MODEL_ASR", "ky-asr-transcribe"), - "subtitle_rendering": os.environ.get("KTV_MODEL_SUBTITLE", "ky-subtitle-render"), - "face_detecting": os.environ.get("KTV_MODEL_FACE", "ky-face-detect"), - "realesrgan_upscaling": os.environ.get("KTV_MODEL_REALESRGAN","ky-realesrgan-upscale"), - "rvc_converting": os.environ.get("KTV_MODEL_RVC", "ky-rvc-convert"), - "video_evaluating": os.environ.get("KTV_MODEL_VIDEO_EVAL","ky-video-eval-evaluate"), - "songrate_evaluating": os.environ.get("KTV_MODEL_SONGRATE", "ky-songrate-evaluate"), + "demucs_separating": os.environ.get("KTV_MODEL_DEMUCS", "demucs-separate"), + "asr_transcribing": os.environ.get("KTV_MODEL_ASR", "asr-transcribe"), + "subtitle_rendering": os.environ.get("KTV_MODEL_SUBTITLE", "subtitle-render"), + "face_detecting": os.environ.get("KTV_MODEL_FACE", "face-detect"), + "realesrgan_upscaling": os.environ.get("KTV_MODEL_REALESRGAN","realesrgan-upscale"), + "rvc_converting": os.environ.get("KTV_MODEL_RVC", "rvc-convert"), + "video_evaluating": os.environ.get("KTV_MODEL_VIDEO_EVAL","video-eval-evaluate"), + "songrate_evaluating": os.environ.get("KTV_MODEL_SONGRATE", "songrate-evaluate"), } LOCAL_WORK_DIR = "/data/pipeline/ktv" @@ -197,7 +197,7 @@ async def handle_video_preparing(tenant_id, task_id, step_name, input_data, conf # ── Demucs ───────────────────────────────────────────────────────── async def handle_demucs_separating(tenant_id, task_id, step_name, input_data, config): - """通过 API 调 ky-demucs-separate。固定使用 separate_full 模式。""" + """通过 API 调 demucs-separate。固定使用 separate_full 模式。""" work_dir = _task_dir(task_id) audio_path = None for dep_output in input_data.values():