diff options
Diffstat (limited to 'app/backends/probe.py')
| -rw-r--r-- | app/backends/probe.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/app/backends/probe.py b/app/backends/probe.py index e996725..64dc7ab 100644 --- a/app/backends/probe.py +++ b/app/backends/probe.py @@ -14,7 +14,8 @@ identity check over plain HTTP: the endpoint names the converter calls (``/run_instruct`` / ``/run_custom_voice`` / ``/generate_custom_voice`` for the CustomVoice demo; ``/run_voice_clone`` / ``/generate_voice_clone`` - for the Base demo). + for the Base demo; ``/run_voice_design`` for the VoiceDesign + demo). ``identify_server`` returns one of the IDENTITY_* constants, or None when the URL does not answer or answers as something unrecognized. It is stdlib-only @@ -33,6 +34,7 @@ IDENTITY_AUDIOCPP = "audiocpp" IDENTITY_FASTER = "faster" IDENTITY_QWEN_CUSTOM = "qwen-custom" IDENTITY_QWEN_CLONE = "qwen-clone" +IDENTITY_QWEN_DESIGN = "qwen-design" # Endpoint names the converter resolves for each qwen demo server (see # converter.clients QwenTTSClient). Mirror them here so identification matches @@ -40,6 +42,7 @@ IDENTITY_QWEN_CLONE = "qwen-clone" _QWEN_CUSTOM_ENDPOINTS = ( "/run_instruct", "/run_custom_voice", "/generate_custom_voice") _QWEN_CLONE_ENDPOINTS = ("/run_voice_clone", "/generate_voice_clone") +_QWEN_DESIGN_ENDPOINTS = ("/run_voice_design",) DEFAULT_TIMEOUT = 3.0 @@ -105,6 +108,8 @@ def _identify_gradio(base: str, timeout: float) -> Optional[str]: return IDENTITY_QWEN_CUSTOM if any(name in endpoints for name in _QWEN_CLONE_ENDPOINTS): return IDENTITY_QWEN_CLONE + if any(name in endpoints for name in _QWEN_DESIGN_ENDPOINTS): + return IDENTITY_QWEN_DESIGN return None |