diff --git a/gradio_app.py b/gradio_app.py
index aa6a71d..fe5a80f 100644
--- a/gradio_app.py
+++ b/gradio_app.py
@@ -149,10 +149,10 @@ def infer_batch(ref_audio, ref_text, gen_text_batches, exp_name, remove_silence,
     generated_waves = []
     spectrograms = []
 
+    if len(ref_text[-1].encode('utf-8')) == 1:
+        ref_text = ref_text + " "
     for i, gen_text in enumerate(progress.tqdm(gen_text_batches)):
         # Prepare the text
-        if len(ref_text[-1].encode('utf-8')) == 1:
-            ref_text = ref_text + " "
         text_list = [ref_text + gen_text]
         final_text_list = convert_char_to_pinyin(text_list)
 
diff --git a/inference-cli.py b/inference-cli.py
index f2fc586..8c292af 100644
--- a/inference-cli.py
+++ b/inference-cli.py
@@ -241,10 +241,10 @@ def infer_batch(ref_audio, ref_text, gen_text_batches, model,ckpt_file,file_voca
     generated_waves = []
     spectrograms = []
 
+    if len(ref_text[-1].encode('utf-8')) == 1:
+        ref_text = ref_text + " "
     for i, gen_text in enumerate(tqdm.tqdm(gen_text_batches)):
         # Prepare the text
-        if len(ref_text[-1].encode('utf-8')) == 1:
-            ref_text = ref_text + " "
         text_list = [ref_text + gen_text]
         final_text_list = convert_char_to_pinyin(text_list)