Update app.py

app.py

@@ -232,56 +232,6 @@ global_audio_list = []
 global_error_list = []
 
 
-def inference_wrapper(
-    text,
-    enable_reference_audio,
-    reference_audio,
-    reference_text,
-    max_new_tokens,
-    chunk_length,
-    top_p,
-    repetition_penalty,
-    temperature,
-    seed,
-    batch_infer_num,
-):
-    audios = []
-    errors = []
-
-    for _ in range(batch_infer_num):
-        result = inference(
-            text,
-            enable_reference_audio,
-            reference_audio,
-            reference_text,
-            max_new_tokens,
-            chunk_length,
-            top_p,
-            repetition_penalty,
-            temperature,
-            seed,
-        )
-
-        _, audio_data, error_message = next(result)
-
-        audios.append(
-            gr.Audio(value=audio_data if audio_data else None, visible=True),
-        )
-        errors.append(
-            gr.HTML(value=error_message if error_message else None, visible=True),
-        )
-
-    for _ in range(batch_infer_num, n_audios):
-        audios.append(
-            gr.Audio(value=None, visible=False),
-        )
-        errors.append(
-            gr.HTML(value=None, visible=False),
-        )
-
-    return None, *audios, *errors
-
-
 def wav_chunk_header(sample_rate=44100, bit_depth=16, channels=1):
     buffer = io.BytesIO()
 
@@ -394,17 +344,14 @@ def build_app():
                                     )
                                 )
 
-                            enable_reference_audio = gr.Checkbox(
-                                label="Enable Reference Audio",
-                            )
-
-                            # Add dropdown for selecting example audio files
-                            example_audio_files = [f for f in os.listdir("examples") if f.endswith(".wav")]
-                            example_audio_dropdown = gr.Dropdown(
-                                label="Select Example Audio",
-                                choices=[""] + example_audio_files,
-                                value=""
-                            )
+                            with gr.Row():
+                                # Add dropdown for selecting example audio files
+                                example_audio_files = [f for f in os.listdir("examples") if f.endswith(".wav")]
+                                example_audio_dropdown = gr.Dropdown(
+                                    label="Select Example Audio",
+                                    choices=[""] + example_audio_files,
+                                    value=""
+                                )
 
                             with gr.Row():
                                 use_memory_cache = gr.Radio(
@@ -418,6 +365,7 @@ def build_app():
                                     label=i18n("Reference Audio"),
                                     type="filepath",
                                 )
+
                             with gr.Row():
                                 reference_text = gr.Textbox(
                                     label=i18n("Reference Text"),
@@ -506,14 +454,14 @@ def build_app():
                 else:
                     lab_content = ""
                 
-                return audio_path, lab_content, True
-            return None, "", False
+                return audio_path, lab_content
+            return None, ""
 
         # Connect the dropdown to update reference audio and text
         example_audio_dropdown.change(
             fn=select_example_audio,
             inputs=[example_audio_dropdown],
-            outputs=[reference_audio, reference_text, enable_reference_audio]
+            outputs=[reference_audio, reference_text]
         )
 
         # Submit
@@ -521,7 +469,7 @@ def build_app():
             inference_wrapper,
             [
                 refined_text,
-                enable_reference_audio,
+                normalize,
                 reference_audio,
                 reference_text,
                 max_new_tokens,