Spaces:

mrfakename
/

DMOSpeech2

Running on Zero

mrfakename commited on 13 days ago

Commit

8517188

1 Parent(s): c246699

fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -96,7 +96,9 @@ def generate_speech(
     if mode == "Custom":
         teacher_steps, teacher_stopping_time, student_start_step = custom_teacher_steps, custom_teacher_stopping_time, custom_student_start_step
     else:
-        teacher_steps, teacher_stopping_time, student_start_step = MODES[mode].values()
     generated_audio = model.generate(
         gen_text=target_text,

     if mode == "Custom":
         teacher_steps, teacher_stopping_time, student_start_step = custom_teacher_steps, custom_teacher_stopping_time, custom_student_start_step
     else:
+        teacher_steps = MODES[mode]["teacher_steps"]
+        teacher_stopping_time = MODES[mode]["teacher_stopping_time"]
+        student_start_step = MODES[mode]["student_start_step"]
     generated_audio = model.generate(
         gen_text=target_text,