Spaces:
Running
Running
feat(tts): add nari-labs/Dia-1.6B model support
Browse files- [feat] Add "nari-labs/Dia-1.6B" to TTS model choices (ui_components.py:447)
- [feat] Implement visibility logic for "nari-labs/Dia-1.6B" model (ui_components.py:520)
- [ui] Adjust default TTS UI element visibility in `else` block (ui_components.py:522)
- ui_components.py +4 -2
ui_components.py
CHANGED
|
@@ -445,7 +445,7 @@ def create_tts_tab(handle_tts_generation_fn):
|
|
| 445 |
with gr.Group():
|
| 446 |
gr.Markdown("**🤖 Model & Provider**")
|
| 447 |
tts_model_name = gr.Dropdown(
|
| 448 |
-
choices=["hexgrad/Kokoro-82M", "ResembleAI/chatterbox"],
|
| 449 |
value=DEFAULT_TTS_MODEL,
|
| 450 |
label="Model",
|
| 451 |
info="Select TTS model"
|
|
@@ -518,8 +518,10 @@ def create_tts_tab(handle_tts_generation_fn):
|
|
| 518 |
return gr.update(visible=True), gr.update(visible=False)
|
| 519 |
elif model_name == "ResembleAI/chatterbox":
|
| 520 |
return gr.update(visible=False), gr.update(visible=True)
|
|
|
|
|
|
|
| 521 |
else:
|
| 522 |
-
return gr.update(visible=
|
| 523 |
|
| 524 |
# Connect model change event
|
| 525 |
tts_model_name.change(
|
|
|
|
| 445 |
with gr.Group():
|
| 446 |
gr.Markdown("**🤖 Model & Provider**")
|
| 447 |
tts_model_name = gr.Dropdown(
|
| 448 |
+
choices=["hexgrad/Kokoro-82M", "ResembleAI/chatterbox", "nari-labs/Dia-1.6B"],
|
| 449 |
value=DEFAULT_TTS_MODEL,
|
| 450 |
label="Model",
|
| 451 |
info="Select TTS model"
|
|
|
|
| 518 |
return gr.update(visible=True), gr.update(visible=False)
|
| 519 |
elif model_name == "ResembleAI/chatterbox":
|
| 520 |
return gr.update(visible=False), gr.update(visible=True)
|
| 521 |
+
elif model_name == "nari-labs/Dia-1.6B":
|
| 522 |
+
return gr.update(visible=False), gr.update(visible=False)
|
| 523 |
else:
|
| 524 |
+
return gr.update(visible=False), gr.update(visible=False)
|
| 525 |
|
| 526 |
# Connect model change event
|
| 527 |
tts_model_name.change(
|