Skip to content

update gradio #98

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
rulerman merged 1 commit into from
Dec 8, 2025
Merged

update gradio #98

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
36 changes: 19 additions & 17 deletions gradio_demo.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -184,9 +184,10 @@ def load_examples_from_jsonl():

# Model configuration
SYSTEM_PROMPT = "You are a speech synthesizer that generates natural, realistic, and human-like conversational audio from dialogue text."
MODEL_PATH = "fnlp/MOSS-TTSD-v0.5"
SPT_CONFIG_PATH = "XY_Tokenizer/config/xy_tokenizer_32k_config.yaml"
SPT_CHECKPOINT_PATH = "XY_Tokenizer/weights/xy_tokenizer.ckpt"
MODEL_PATH = "fnlp/MOSS-TTSD-v0.7"
# Align SPT config/weights with CLI inference
SPT_CONFIG_PATH = "XY_Tokenizer/config/MOSS_TTSD_tokenizer.yaml"
SPT_CHECKPOINT_PATH = "XY_Tokenizer/weights/MOSS_TTSD_tokenizer"
MAX_CHANNELS = 8

# Global variables for caching loaded models
Expand Down Expand Up @@ -251,29 +252,29 @@ def process_single_audio_generation(

# Handle different audio input modes (mutually exclusive)
if audio_mode == "Single":
# Use single audio mode
# Strict single-audio requirement
if not prompt_audio_single:
return (
None,
"Error: In Single mode, please provide one prompt_audio and its text.",
)
item["prompt_audio"] = prompt_audio_single
item["prompt_text"] = prompt_text_single
elif audio_mode == "Role" and prompt_audio_1 and prompt_audio_2:
# Use role audio mode (requires both audio files)
elif audio_mode == "Role":
# Strict role-audio requirement: both speakers must be provided
if not (prompt_audio_1 and prompt_audio_2):
return (
None,
"Error: In Role mode, please provide both Role1 and Role2 reference audios.",
)
item["prompt_audio_speaker1"] = prompt_audio_1
item["prompt_text_speaker1"] = prompt_text_1 if prompt_text_1 else ""
item["prompt_audio_speaker2"] = prompt_audio_2
item["prompt_text_speaker2"] = prompt_text_2 if prompt_text_2 else ""
elif audio_mode == "Role" and prompt_audio_1:
# Only Role 1 audio provided, treat as single audio
print("Only Role 1 audio provided, treating as single audio.")
item["prompt_audio"] = prompt_audio_1
item["prompt_text"] = prompt_text_1 if prompt_text_1 else ""
elif audio_mode == "Role" and prompt_audio_2:
# Only Role 2 audio provided, treat as single audio
print("Only Role 2 audio provided, treating as single audio.")
item["prompt_audio"] = prompt_audio_2
item["prompt_text"] = prompt_text_2 if prompt_text_2 else ""
else:
return (
None,
"Error: Please select a mode and provide corresponding audio files\n- Single Audio Mode: Provide one audio file and corresponding text\n- Role Mode: Provide audio files for Role1 and Role2",
"Error: Please select an audio input mode (Single or Role).",
)

# Set random seed to ensure reproducible results
Expand All @@ -290,6 +291,7 @@ def process_single_audio_generation(
system_prompt=SYSTEM_PROMPT,
start_idx=0,
use_normalize=use_normalize,
silence_duration=0.1,
)

# Check results
Expand Down